Releases: jax-ml/jax
Releases · jax-ml/jax
Jax release v0.3.25
jax-v0.3.25 jax version 0.3.25
Jaxlib release v0.3.24
jaxlib-v0.3.24 jaxlib version 0.3.24
Jax release v0.3.24
jax-v0.3.24 jax version 0.3.24
jax-v0.3.23
- Changes
- Update Colab TPU driver version for new jaxlib release.
jaxlib-v0.3.22
jaxlib version 0.3.22
jax-v0.3.22
- Changes
- Add
JAX_PLATFORMS=tpu,cpu
as default setting in TPU initialization,
so JAX will raise an error if TPU cannot be initialized instead of falling
back to CPU. SetJAX_PLATFORMS=''
to override this behavior and automatically
choose an available backend (the original default), or setJAX_PLATFORMS=cpu
to always use CPU regardless of if the TPU is available.
- Add
- Deprecations
- Several test utilities deprecated in JAX v0.3.8 are now removed from
{mod}jax.test_util
.
- Several test utilities deprecated in JAX v0.3.8 are now removed from
JAX release v0.3.21
- Changes
- The persistent compilation cache will now warn instead of raising an
exception on error ({jax-issue}#12582
), so program execution can continue
if something goes wrong with the cache. Set
JAX_RAISE_PERSISTENT_CACHE_ERRORS=true
to revert this behavior.
- The persistent compilation cache will now warn instead of raising an
jaxlib release v0.3.20
Notable changes:
- Fixes support for limiting the visible CUDA devices via
jax_cuda_visible_devices
in distributed jobs. This functionality is needed for the JAX/SLURM integration on GPU (#12533).
JAX release v0.3.20
JAX release v0.3.19
Fixes the required jaxlib version