Jax: jax-v0.4.28 Release

Release date:
May 9, 2024
Previous version:
jax-v0.4.28-rc (released May 9, 2024)
Magnitude:
0 Diff Delta
Contributors:
0 total committers
Data confidence:
Commits:

Top Contributors in jax-v0.4.28

Could not determine top contributors for this release.

Directory Browser for jax-v0.4.28

We haven't yet finished calculating and confirming the files and directories changed in this release. Please check back soon.

Release Notes Published

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