Bump jax from 0.8.2 to 0.9.0 (#1487)

Bumps [jax](https://github.com/jax-ml/jax) from 0.8.2 to 0.9.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.8.2...jax-v0.9.0)

---
updated-dependencies:
- dependency-name: jax
  dependency-version: 0.9.0
  dependency-type: direct:development
  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>

Author: dependabot[bot]

requirements-dev.txt

@@ -15,7 +15,7 @@ fsspec==2026.1.0 ; python_version >= "3.11" and python_version < "3.15"
 healpy==1.19.0 ; python_version >= "3.11" and python_version < "3.15"
 idna==3.11 ; python_version >= "3.11" and python_version < "3.15"
 iniconfig==2.3.0 ; python_version >= "3.11" and python_version < "3.15"
-jax==0.8.2 ; python_version >= "3.11" and python_version < "3.15"
+jax==0.9.0 ; python_version >= "3.11" and python_version < "3.15"
 jaxlib==0.9.0 ; python_version >= "3.11" and python_version < "3.15"
 jinja2==3.1.6 ; python_version >= "3.11" and python_version < "3.15"
 kiwisolver==1.4.9 ; python_version >= "3.11" and python_version < "3.15"