pybamm-team#3049 pybamm-team#3121 sync jax, jaxlib version requirements

js1tr3 · Aug 12, 2024 · ef9fb06 · ef9fb06
1 parent d8a7cfe
commit ef9fb06
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -105,10 +105,11 @@ dev = [
 pandas =  [
     "pandas>=0.24",
 ]
-# For the Jax solver
+# For the Jax solver. Note: these should be kept in sync with the versions defined
+# in noxfile.py and pybamm/util.py.
 jax = [
-    "jax==0.4.8",
-    "jaxlib==0.4.7",
+    "jax>=0.4,<=0.5",
+    "jaxlib>=0.4,<=0.5",
 ]
 # For the scikits.odes solver
 odes = [