From 325bc5a057798d3bdd2210579d1664a11c78bb20 Mon Sep 17 00:00:00 2001 From: Brendt Wohlberg Date: Wed, 13 Dec 2023 09:27:30 -0700 Subject: [PATCH] Bump max supported jaxlib/jax version --- CHANGES.rst | 2 +- requirements.txt | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/CHANGES.rst b/CHANGES.rst index 2d68d754a..d3f1c9bf2 100644 --- 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 index 8a44a695f..5ac3c6904 100644 --- 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