# JAX v0.4.28 — JAX v0.4.28

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

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** — Reverted a change to make_jaxpr that was breaking Equinox
- **removed** — The kind argument to jax.numpy.sort and jax.numpy.argsort is now removed, use stable=True or stable=False instead
- **removed** — Removed get_compute_capability from the jax.experimental.pallas.gpu module, use the compute_capability attribute of a GPU device instead
- **changed** — The minimum jaxlib version of this release is 0.4.27

* Bug fixes
  * Reverted a change to `make_jaxpr` that was breaking Equinox (#21116).

* Deprecations & removals
  * The ``kind`` argument to `jax.numpy.sort` and `jax.numpy.argsort`
    is now removed. Use `stable=True` or `stable=False` instead.
  * Removed ``get_compute_capability`` from the ``jax.experimental.pallas.gpu``
    module. Use the ``compute_capability`` attribute of a GPU device, returned
    by `jax.devices` or `jax.local_devices`, instead.

* Changes
  * The minimum jaxlib version of this release is 0.4.27.