#3959 use jax and jaxlib 0.4.27

pybamm-team · May 20, 2024 · 3abc482 · 3abc482
1 parent a01daab
commit 3abc482
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -113,8 +113,8 @@ dev = [
 ]
 # For the Jax solver. Note: these must be kept in sync with the versions defined in pybamm/util.py.
 jax = [
-    "jax==0.4.28",
-    "jaxlib==0.4.28",
+    "jax==0.4.27",
+    "jaxlib==0.4.27",
 ]
 # Contains all optional dependencies, except for jax and dev dependencies
 all = [