diff --git a/CHANGES.rst b/CHANGES.rst index ee8c8c10..a32c3854 100644 --- a/CHANGES.rst +++ b/CHANGES.rst @@ -18,7 +18,7 @@ Version 0.0.6 (unreleased) • Rename ``scico.flax.save_weights`` and ``scico.flax.load_weights`` to ``scico.flax.save_variables`` and ``scico.flax.load_variables`` respectively. -• Support ``jaxlib`` and ``jax`` versions 0.4.3 to 0.4.30. +• Support ``jaxlib`` and ``jax`` versions 0.4.3 to 0.4.31. • Support ``flax`` versions 0.8.0 to 0.8.3. diff --git a/requirements.txt b/requirements.txt index cb78a313..8bfc6f16 100644 --- 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.30 -jax>=0.4.3,<=0.4.30 +jaxlib>=0.4.3,<=0.4.31 +jax>=0.4.3,<=0.4.31 orbax-checkpoint<=0.5.7 flax>=0.8.0,<=0.8.3 pyabel>=0.9.0