Update pyproject.toml

YaoGroup · Jul 26, 2024 · 54dfa58 · 54dfa58
1 parent 7f0834b
commit 54dfa58
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -13,6 +13,14 @@ classifiers = [
     "Operating System :: OS Independent",
 ]
 
+dependencies = [
+  "jax >= 0.4.20",
+  "jaxlib >= 0.4.20",
+  "tensorflow-probability >= 0.23.0",
+  "matplotlib >= 3.8.0",
+  "optax >= 0.1.7"
+]
+
 [project.urls]
 Homepage = "https://github.com/YaoGroup/DIFFICE_jax"
 Issues = "https://github.com/YaoGroup/DIFFICE_jax/issues"