Releases: jax-ml/jax
Releases · jax-ml/jax
JAX v0.4.28
-
Bug fixes
- Reverted a change to
make_jaxpr
that was breaking Equinox (#21116).
- Reverted a change to
-
Deprecations & removals
- The
kind
argument tojax.numpy.sort
andjax.numpy.argsort
is now removed. Usestable=True
orstable=False
instead. - Removed
get_compute_capability
from thejax.experimental.pallas.gpu
module. Use thecompute_capability
attribute of a GPU device, returned
byjax.devices
orjax.local_devices
, instead.
- The
-
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
Jaxlib release v0.4.23
jaxlib-v0.4.23 jaxlib version 0.4.23