Skip to content

jaxlib release v0.3.20

Choose a tag to compare

View all tags
@hawkinsp hawkinsp released this 28 Sep 18:07
· 23276 commits to main since this release
jaxlib-v0.3.20
0282b4b

Notable changes:

  • Fixes support for limiting the visible CUDA devices viajax_cuda_visible_devices in distributed jobs. This functionality is needed for the JAX/SLURM integration on GPU (#12533).
Assets 2
Loading