Bump maximum jax version

lanl · Mar 3, 2024 · 4a79e3f · 4a79e3f
1 parent e43c4c4
commit 4a79e3f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -4,8 +4,8 @@ scipy>=1.6.0
 imageio>=2.17
 tifffile
 matplotlib
-jaxlib>=0.4.3,<=0.4.24
-jax>=0.4.3,<=0.4.24
+jaxlib>=0.4.3,<=0.4.25
+jax>=0.4.3,<=0.4.25
 orbax-checkpoint
 flax>=0.6.1,<=0.7.5
 pyabel>=0.9.0