From a28da46fca57902b21490b6eef128b4afd815855 Mon Sep 17 00:00:00 2001
From: Brendt Wohlberg <bwohlberg@users.noreply.github.com>
Date: Tue, 30 Jul 2024 12:04:20 -0600
Subject: [PATCH] Bump max supported jax version (#546)

---
 CHANGES.rst      | 2 +-
 requirements.txt | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/CHANGES.rst b/CHANGES.rst
index 4be84b0c..bdcfd129 100644
--- a/CHANGES.rst
+++ b/CHANGES.rst
@@ -16,7 +16,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