diff --git a/CHANGES.rst b/CHANGES.rst index 472c777e..6c558758 100644 --- 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.*. diff --git a/requirements.txt b/requirements.txt index 838a6b18..1f67e183 100644 --- 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.* pyabel>=0.9.0