chore: update jax requirement from <=0.5.0,>=0.4.28 to >=0.4.28,<=0.5.1

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.4.28...jax-v0.5.1)

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

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

pyproject.toml

@@ -31,7 +31,7 @@ dependencies = [
     "h5py<=3.13.0",
     "iminuit<=2.30.1",
     "jax>=0.4.28; python_version < '3.10'",
-    "jax>=0.4.28,<=0.5.0; python_version >= '3.10'",
+    "jax>=0.4.28,<=0.5.1; python_version >= '3.10'",
     "jaxns==2.6.7",
     "matplotlib",
     "nautilus-sampler==1.0.5",