upper pin for jax, fixes #213

liesel-devs · Sep 20, 2024 · 24d583d · 24d583d
1 parent df0d6fb
commit 24d583d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.cfg b/setup.cfg
@@ -21,8 +21,8 @@ install_requires =
     blackjax>=1.0
     deprecated>=1.2
     dill>=0.3
-    jax>=0.4.1
-    jaxlib>=0.4.1
+    jax>=0.4.1,<=0.4.31
+    jaxlib>=0.4.1,<=0.4.31
     matplotlib>=3.5
     networkx>=2.6
     numpy>=1.22,!=1.24.0,<2.0