⬆️ Bump jax[cuda12] from 0.4.38 to 0.6.1

Bumps [jax[cuda12]](https://github.com/jax-ml/jax) from 0.4.38 to 0.6.1.
- [Release notes](https://github.com/jax-ml/jax/releases)
- [Changelog](https://github.com/jax-ml/jax/blob/main/CHANGELOG.md)
- [Commits](jax-ml/jax@jax-v0.4.38...jax-v0.6.1)

---
updated-dependencies:
- dependency-name: jax[cuda12]
  dependency-version: 0.6.1
  dependency-type: direct:development
  update-type: version-update:semver-minor
...

Signed-off-by: dependabot[bot] <support@github.com>

Author: dependabot[bot]

setup.py

@@ -30,7 +30,7 @@
     author_email='chao.brain@qq.com',
     packages=packages,
     python_requires='>=3.10',
-    install_requires=['numpy>=1.15', 'jax==0.4.38', 'tqdm'],
+    install_requires=['numpy>=1.15', 'jax==0.6.1', 'tqdm'],
     url='https://github.com/brainpy/BrainPy',
     project_urls={
         "Bug Tracker": "https://github.com/brainpy/BrainPy/issues",