From 806710a01d6f1e40c6f5c8377a9f5ea87d5764b4 Mon Sep 17 00:00:00 2001
From: Brendt Wohlberg <brendt@ieee.org>
Date: Wed, 11 Dec 2024 07:07:08 -0700
Subject: [PATCH] Bump jax and flax version upper bounds

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

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