jax-v0.4.28

版本发布时间: 2024-05-10 07:28:46

jax-ml/jax最新发布版本:jaxlib-v0.4.32(2024-09-12 04:03:04)

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.