jaxlib v0.4.28

• Bug fixes

  • Fixes a memory corruption bug in the type name of Array and JIT Python
    objects in Python 3.10 or earlier.
  • Fixed a warning '+ptx84' is not a recognized feature for this target
    under CUDA 12.4.
  • Fixed a slow compilation problem on CPU.

• Changes

  • The Windows build is now built with Clang instead of MSVC.

JAX v0.4.28

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

Jaxlib release v0.4.27

jaxlib-v0.4.27

jaxlib version 0.4.27

Jax release v0.4.27

jax-v0.4.27

jax version 0.4.27

Jaxlib release v0.4.26

jaxlib-v0.4.26

jaxlib version 0.4.26

Jax release v0.4.26

jax-v0.4.26

jax version 0.4.26

Jaxlib release v0.4.25

jaxlib-v0.4.25

jaxlib version 0.4.25

JAX release v0.4.25

jax-v0.4.25

jax version 0.4.25

jaxlib release v0.4.24

jaxlib release v0.4.24

JAX release v0.4.24

JAX release v0.4.24