# JAX v0.5.3 — JAX v0.5.3

- Product: JAX (https://whatsnew.fyi/product/jax)
- Vendor: Google
- Date: 2025-03-19
- Version: v0.5.3
- Original notes: https://github.com/jax-ml/jax/releases/tag/jax-v0.5.3
- Permalink: https://whatsnew.fyi/product/jax/releases/v0.5.3

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

---

- **added** — Added a allow_negative_indices option to jax.lax.dynamic_slice, jax.lax.dynamic_update_slice and related functions to optionally disable code generation for clamping negative indices
- **added** — Added a replace option to jax.random.categorical to enable sampling without replacement

* New Features

  * Added a `allow_negative_indices` option to `jax.lax.dynamic_slice`,
    `jax.lax.dynamic_update_slice` and related functions. The default is
    true, matching the current behavior. If set to false, JAX does not need to
    emit code clamping negative indices, which improves code size.
  * Added a `replace` option to `jax.random.categorical` to enable sampling
    without replacement.