Bump jax from 0.4.38 to 0.5.0 (#695)

Bumps [jax](https://github.com/jax-ml/jax) from 0.4.38 to 0.5.0.
- [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.5.0)

---
updated-dependencies:
- dependency-name: jax
  dependency-type: direct:production
  update-type: version-update:semver-minor
...

Signed-off-by: dependabot[bot] <[email protected]>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>

requirements_jax.txt

@@ -1 +1 @@
-jax==0.4.38
+jax==0.5.0