Update jax requirement from <=0.4.36,>=0.3.2 to >=0.3.2,<=0.4.37

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](jax-ml/jax@jax-v0.3.2...jax-v0.4.37)

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

Signed-off-by: dependabot[bot] <[email protected]>

requirements.txt

@@ -1,3 +1,3 @@
 equinox >= 0.11.0, <= 0.11.10
-jax >= 0.3.2, <= 0.4.36
+jax >= 0.3.2, <= 0.4.37
 numpy >= 1.20.0, <= 2.1.3