build: pinning jax to v0.4.7 (#156)

peng-lab · Nov 12, 2024 · 4b75d3f · 4b75d3f
1 parent 04fb477
commit 4b75d3f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.cfg b/setup.cfg
@@ -23,8 +23,8 @@ project_urls =
 packages = find:
 install_requires =
     hyperactive>=4.4.0
-    jax>=0.3.10,<=0.4.23
-    jaxlib>=0.3.10,<=0.4.23
+    jax>=0.4.7,<=0.4.7
+    jaxlib>=0.4.7,<=0.4.7
     numpy
     pooch
     pydantic>=2.7.0,<3.0.0