From e3b68f56cdc1a2336fa5ecbb10fa9f758945a347 Mon Sep 17 00:00:00 2001
From: "dependabot[bot]" <49699333+dependabot[bot]@users.noreply.github.com>
Date: Mon, 22 Dec 2025 14:10:23 +0000
Subject: [PATCH] Update jax requirement from <0.6.1,>=0.4.25 to
 >=0.4.25,<0.8.3

Updates the requirements on [jax](https://github.com/jax-ml/jax) to permit the latest version.
- [Release notes](https://github.com/jax-ml/jax/releases)
- [Changelog](https://github.com/jax-ml/jax/blob/main/CHANGELOG.md)
- [Commits](https://github.com/jax-ml/jax/compare/jax-v0.4.25...jax-v0.8.2)

---
updated-dependencies:
- dependency-name: jax
  dependency-version: 0.8.2
  dependency-type: direct:production
...

Signed-off-by: dependabot[bot] <support@github.com>
---
 pyproject.toml   | 2 +-
 requirements.txt | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index cd65095..bfaef5f 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -26,7 +26,7 @@ dependencies = [
     "qnm==0.4.4",
     "seaborn~=0.13",
     "numpyro==0.19.0",
-    "jax>=0.4.25,<0.6.1",
+    "jax>=0.4.25,<0.8.3",
     "parse>=1.20",
     "jaxlib==0.4.38; sys_platform == 'darwin' and platform_machine == 'x86_64'",
     "tables~=3.10",
diff --git a/requirements.txt b/requirements.txt
index f4237b1..0463f3f 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -9,7 +9,7 @@ jupyterlab
 ipywidgets
 ipykernel
 numpyro~=0.17
-jax>=0.4.25,<0.6.1
+jax>=0.4.25,<0.8.3
 parse>=1.20
 jaxlib==0.4.38; sys_platform == 'darwin' and platform_machine == 'x86_64'
 tables~=3.10