Package profile
jax-cuda12-plugin
- Summary: JAX Plugin for NVIDIA GPUs
- Author: JAX team
- Homepage: https://github.com/jax-ml/jax
- Source: https://github.com/jax-ml/jax (Repo profile)
- Number of releases: 22
- First release: 0.4.22 on 2023-12-14
- Latest release: 0.6.2 on 2025-06-17
Releases
PyPI Downloads
Dependencies
Jax-cuda12-plugin has 13 dependencies, 12 of which optional.| Dependency | Optional |
|---|---|
| jax-cuda12-pjrt | false |
| nvidia-cublas-cu12 | true |
| nvidia-cuda-cupti-cu12 | true |
| nvidia-cuda-nvcc-cu12 | true |
| nvidia-cuda-nvrtc-cu12 | true |
| nvidia-cuda-runtime-cu12 | true |
| nvidia-cudnn-cu12 | true |
| nvidia-cufft-cu12 | true |
| nvidia-cusolver-cu12 | true |
| nvidia-cusparse-cu12 | true |
| nvidia-nccl-cu12 | true |
| nvidia-nvjitlink-cu12 | true |
| nvidia-nvshmem-cu12 | true |