JAX v0.4.28

v0.4.28

JAX v0.4.28

Changed 1
  • The minimum jaxlib version of this release is 0.4.27
Fixed 1
  • Reverted a change to make_jaxpr that was breaking Equinox
Removed 2
  • 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 instead

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

Upgraded? How did it go?

Discussion