#jax

1 bookmark 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]