#jax

1bookmark tagged with "jax"

across 1 category: gpu-nvidia

• JAX CUDA 13 Support

  PyPI • Aug 5, 2025 • gpu-nvidia

  JAX 0.7.0+ includes official CUDA 13 support via pip extras, making it the first major ML framework to support CUDA 13

  [jax][cuda][cuda-13][machine-learning][gpu]