# JAX v0.9.0.1 — JAX v0.9.0.1

- Product: JAX (https://whatsnew.fyi/product/jax)
- Vendor: Google
- Date: 2026-02-05
- Version: v0.9.0.1
- Original notes: https://github.com/jax-ml/jax/releases/tag/jax-v0.9.0.1
- Permalink: https://whatsnew.fyi/product/jax/releases/v0.9.0.1

What's New is an index, not a publisher: every entry below links to the vendor's own release notes, which are the authoritative source. Entries are labelled where they are hand-curated sample data, pre-releases, or drawn from a secondary source such as a developer blog.

Reuse: the summaries, labels and curation here are © What's New. Quote freely with attribution and a link back; wholesale republication of the corpus is not permitted — terms: https://whatsnew.fyi/terms. The vendors' own release notes remain their publishers'.

---

JAX v0.9.0.1 is identical to v0.9.0 with the commits from the following four PRs patched in:

- https://github.com/openxla/xla/pull/36579
- https://github.com/openxla/xla/pull/36345
- https://github.com/openxla/xla/pull/36755
- https://github.com/openxla/xla/pull/36696