# JAX v0.4.37 — JAX v0.4.37 - Product: JAX (https://whatsnew.fyi/product/jax) - Vendor: Google - Date: 2024-12-10 - Version: v0.4.37 - Original notes: https://github.com/jax-ml/jax/releases/tag/jax-v0.4.37 - Permalink: https://whatsnew.fyi/product/jax/releases/v0.4.37 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'. --- - **fixed** — Fixed a bug where jit would error if an argument was named f - **fixed** — Fixed a bug that throws index out of range error in jax.lax.while_loop if the user registers pytree node class with different aux data for the flatten and flatten_with_path - **fixed** — Pinned libtpu release 0.0.6 that fixes a compiler bug on TPU v6e This is a patch release of jax 0.4.36. Only "jax" was released at this version. * Bug fixes * Fixed a bug where `jit` would error if an argument was named `f` (#25329). * Fix a bug that will throw `index out of range` error in `jax.lax.while_loop` if the user registers pytree node class with different aux data for the flatten and flatten_with_path. * Pinned a new libtpu release (0.0.6) that fixes a compiler bug on TPU v6e.