pybamm-team#3443 Bump to latest version of jax and jaxlib

tested with `--upgrade` and `--upgrade-strategy eager` plus `--no-cache-dir`
js1tr3 · Aug 12, 2024 · 913346f · 913346f
1 parent 09e00ba
commit 913346f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -389,8 +389,8 @@ def compile_KLU():
             # via the PyBaMM images on Docker Hub which come with
             # Python 3.11 installed.
             # It also provides support for CPU-only Jax on Windows.
-            "jax==0.4.18; python_version >= '3.9'",
-            "jaxlib==0.4.18; python_version >= '3.9'",
+            "jax==0.4.20; python_version >= '3.9'",
+            "jaxlib==0.4.20; python_version >= '3.9'",
             # Jax 0.4.13 was the last version to support Python 3.8.
             # Support for CPU-only Windows was added in 0.4.13, so
             # this version supports Windows too.