Bump max supported jaxlib/jax version

lanl · Dec 13, 2023 · 325bc5a · 325bc5a
1 parent 98ebb89
commit 325bc5a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/CHANGES.rst b/CHANGES.rst
@@ -17,7 +17,7 @@ Version 0.0.5   (unreleased)
 • Rename ``solver.ATADSolver`` to ``solver.MatrixATADSolver``.
 • Rename some ``__init__`` parameters of ``linop.DiagonalStack`` and
   ``linop.VerticalStack``.
-• Support ``jaxlib`` and ``jax`` versions 0.4.3 to 0.4.20.
+• Support ``jaxlib`` and ``jax`` versions 0.4.3 to 0.4.21.
 • Support ``flax`` versions up to 0.7.5.
 • Use ``orbax`` for checkpointing ``flax`` models.
 

diff --git a/requirements.txt b/requirements.txt
@@ -3,8 +3,8 @@ scipy>=1.6.0
 tifffile
 imageio>=2.17
 matplotlib
-jaxlib>=0.4.3,<=0.4.20
-jax>=0.4.3,<=0.4.20
+jaxlib>=0.4.3,<=0.4.21
+jax>=0.4.3,<=0.4.21
 flax>=0.6.1,<=0.7.5
 svmbir>=0.3.3
 pyabel>=0.9.0