jaxlib release v0.3.20
hawkinsp
released this
28 Sep 18:07
·
10060 commits
to main
since this release
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).