From 9f7a75bfd4edd7d07192e60e47317288c33ee337 Mon Sep 17 00:00:00 2001
From: "dependabot[bot]" <49699333+dependabot[bot]@users.noreply.github.com>
Date: Fri, 22 Sep 2023 01:41:54 +0000
Subject: [PATCH] :arrow_up: Update jax requirement

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

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

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

diff --git a/requirements-dev.txt b/requirements-dev.txt
index 49fa49722..67dc27f4b 100644
--- a/requirements-dev.txt
+++ b/requirements-dev.txt
@@ -1,7 +1,7 @@
 numpy
 numba
 brainpylib
-jax>=0.4.1, <0.4.16
+jax>=0.4.1, <0.4.17
 jaxlib>=0.4.1, <0.4.16
 matplotlib>=3.4
 msgpack
diff --git a/requirements-doc.txt b/requirements-doc.txt
index e6e498937..b3780a5ce 100644
--- a/requirements-doc.txt
+++ b/requirements-doc.txt
@@ -2,7 +2,7 @@ numpy
 tqdm
 msgpack
 numba
-jax>=0.4.1, <0.4.16
+jax>=0.4.1, <0.4.17
 jaxlib>=0.4.1, <0.4.16
 matplotlib>=3.4
 scipy>=1.1.0
diff --git a/requirements.txt b/requirements.txt
index ebf85b86e..eddace704 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,5 @@
 numpy
-jax>=0.4.1, <0.4.16
+jax>=0.4.1, <0.4.17
 tqdm
 msgpack
 numba
\ No newline at end of file
diff --git a/setup.py b/setup.py
index 68debcdee..b1cc12297 100644
--- a/setup.py
+++ b/setup.py
@@ -51,7 +51,7 @@
   author_email='chao.brain@qq.com',
   packages=packages,
   python_requires='>=3.8',
-  install_requires=['numpy>=1.15', 'jax>=0.4.1, <0.4.16', 'tqdm', 'msgpack', 'numba'],
+  install_requires=['numpy>=1.15', 'jax>=0.4.1, <0.4.17', 'tqdm', 'msgpack', 'numba'],
   url='https://github.com/brainpy/BrainPy',
   project_urls={
     "Bug Tracker": "https://github.com/brainpy/BrainPy/issues",