Package profile
jax-cuda12-plugin
- Summary: JAX Plugin for NVIDIA GPUs
- Author: JAX team
- License: Apache-2.0
- Homepage: https://github.com/jax-ml/jax
- Source: https://github.com/jax-ml/jax
- Number of releases: 41
- First release: 0.4.22 on 2023-12-14
- Latest release: 0.12.0.1 on 2026-10-21
- Latest release size: 8.4 MB (wheel)