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

tested with `--upgrade` and `--upgrade-strategy eager` plus `--no-cache-dir`
agriyakhetarpal · Nov 22, 2023 · e17163f · e17163f
1 parent e2d849d
commit e17163f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -272,8 +272,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.