included minimum requirements

AaltoML · Aug 29, 2023 · 59f52fc · 59f52fc
1 parent 339e986
commit 59f52fc
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -7,6 +7,13 @@
     version=__version__,
     packages=find_packages(),
     python_requires='>=3.6',
+    install_requires=[
+        "jax==0.4.14",
+        "jaxlib==0.4.14",
+        "objax==1.7.0",
+        "tensorflow_probability==0.20.1",
+        "numpy>=1.22"
+    ],
     url='https://github.com/AaltoML/BayesNewton',
     license='Apache-2.0',
 )