Bump jax from 0.4.31 to 0.4.32

Bumps [jax](https://github.com/google/jax) from 0.4.31 to 0.4.32.
- [Release notes](https://github.com/google/jax/releases)
- [Changelog](https://github.com/google/jax/blob/main/CHANGELOG.md)
- [Commits](jax-ml/jax@jax-v0.4.31...jax-v0.4.32)

---
updated-dependencies:
- dependency-name: jax
  dependency-type: direct:development
  update-type: version-update:semver-patch
...

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