jax

124c723a - Prefer binaries in NVIDIA `nvcc` wheel over system CUDA installation in Mosaic GPU implementation.