Testing jax and jaxlib versions #2

maurorigo · Sep 6, 2024 · c6c9daf · c6c9daf
1 parent 6a681e0
commit c6c9daf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -112,8 +112,8 @@ def build_extension(self, ext):
     package_dir={"": "src"},
     include_package_data=True,
     install_requires=[
-        "jax>=0.4.16,<=0.4.23",
-        "jaxlib>=0.4.16,<=0.4.23",
+        "jax>=0.4.16,<=0.4.26",
+        "jaxlib>=0.4.16,<=0.4.26",
         "numpy<2",
         "mpi4jax>=0.4.0"
     ],