# JAX v0.4.29 — JAX v0.4.29

- Product: JAX (https://whatsnew.fyi/product/jax)
- Vendor: Google
- Date: 2024-06-10
- Version: v0.4.29
- Original notes: https://github.com/jax-ml/jax/releases/tag/jax-v0.4.29
- Permalink: https://whatsnew.fyi/product/jax/releases/v0.4.29

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'.

---

- **changed** — JAX now requires ml_dtypes version 0.4.0 or newer
- **removed** — Removed backwards-compatibility support for old usage of the jax.experimental.export API; use 'from jax.experimental import export' instead of 'from jax.experimental.export import export'
- **deprecated** — jax.sharding.XLACompatibleSharding is deprecated; use jax.sharding.Sharding instead
- **deprecated** — jax.experimental.Exported.in_shardings has been renamed to jax.experimental.Exported.in_shardings_hlo; same for out_shardings
- **removed** — Removed jax.core.non_negative_dim, jax.core.DimSize, and jax.core.Shape
- **removed** — Removed jax.lax.tie_in
- **removed** — Removed jax.nn.normalize
- **removed** — Removed jax.interpreters.xla.backend_specific_translations, jax.interpreters.xla.translations, jax.interpreters.xla.register_translation, jax.interpreters.xla.xla_destructure, jax.interpreters.xla.TranslationRule, jax.interpreters.xla.TranslationContext, and jax.interpreters.xla.XlaOp
- **deprecated** — The tol argument of jax.numpy.linalg.matrix_rank is being deprecated; use rtol instead
- **deprecated** — The rcond argument of jax.numpy.linalg.pinv is being deprecated; use rtol instead
- **removed** — Removed the deprecated jax.config submodule; use jax.config directly via 'import jax' instead
- **changed** — jax.random APIs no longer accept batched keys; use jax.vmap explicitly in such cases
- **added** — Added jax.experimental.Exported.in_shardings_jax to construct shardings that can be used with the JAX APIs from the HloShardings stored in Exported objects

* Changes
  * We anticipate that this will be the last release of JAX and jaxlib
    supporting a monolithic CUDA jaxlib. Future releases will use the CUDA
    plugin jaxlib (e.g. `pip install jax[cuda12]`).
  * JAX now requires ml_dtypes version 0.4.0 or newer.
  * Removed backwards-compatibility support for old usage of the
    `jax.experimental.export` API. It is not possible anymore to use
    `from jax.experimental.export import export`, and instead you should use
    `from jax.experimental import export`.
    The removed functionality has been deprecated since 0.4.24.

* Deprecations
  * `jax.sharding.XLACompatibleSharding` is deprecated. Please use
    `jax.sharding.Sharding`.
  * `jax.experimental.Exported.in_shardings` has been renamed as
    `jax.experimental.Exported.in_shardings_hlo`. Same for `out_shardings`.
    The old names will be removed after 3 months.
  * Removed a number of previously-deprecated APIs:
    * from {mod}`jax.core`: `non_negative_dim`, `DimSize`, `Shape`
    * from {mod}`jax.lax`: `tie_in`
    * from {mod}`jax.nn`: `normalize`
    * from {mod}`jax.interpreters.xla`: `backend_specific_translations`,
      `translations`, `register_translation`, `xla_destructure`,
      `TranslationRule`, `TranslationContext`, `XlaOp`.
  * The ``tol`` argument of {func}`jax.numpy.linalg.matrix_rank` is being
    deprecated and will soon be removed. Use `rtol` instead.
  * The ``rcond`` argument of {func}`jax.numpy.linalg.pinv` is being
    deprecated and will soon be removed. Use `rtol` instead.
  * The deprecated `jax.config` submodule has been removed. To configure JAX
    use `import jax` and then reference the config object via `jax.config`.
  * {mod}`jax.random` APIs no longer accept batched keys, where previously
    some did unintentionally. Going forward, we recommend explicit use of
    {func}`jax.vmap` in such cases.

* New Functionality
  * Added {func}`jax.experimental.Exported.in_shardings_jax` to construct
    shardings that can be used with the JAX APIs from the HloShardings
    that are stored in the `Exported` objects.