Bump jax and flax version upper bounds (#572)

* Bump jax and flax version upper bounds * Fix requirements syntax error
lanl · Dec 11, 2024 · 49a6ebb · 49a6ebb
1 parent 20899df
commit 49a6ebb
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/CHANGES.rst b/CHANGES.rst
@@ -7,6 +7,8 @@ Version 0.0.7   (unreleased)
 ----------------------------
 
 • New module ``scico.trace`` for tracing function/method calls.
+• Support ``jaxlib`` and ``jax`` versions 0.4.13 to 0.4.37.
+• Support ``flax`` versions 0.8.0 to 0.10.2.
 
 
 

diff --git a/requirements.txt b/requirements.txt
@@ -4,8 +4,8 @@ scipy>=1.6.0
 imageio>=2.17
 tifffile
 matplotlib
-jaxlib>=0.4.13,<=0.4.35
-jax>=0.4.13,<=0.4.35
+jaxlib>=0.4.13,<=0.4.37
+jax>=0.4.13,<=0.4.37
 orbax-checkpoint>=0.5.0
-flax>=0.8.0,<=0.10.0
+flax>=0.8.0,<=0.10.2
 pyabel>=0.9.0
-Original file line number
+Diff line change
@@ Expand Up / @@ -7,6 +7,8 @@ Version 0.0.7 (unreleased) @@
     ----------------------------
     • New module ``scico.trace`` for tracing function/method calls.
+    • Support ``jaxlib`` and ``jax`` versions 0.4.13 to 0.4.37.
+    • Support ``flax`` versions 0.8.0 to 0.10.2.
@@ Expand Down @@