Package search
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: 21
- First release: 0.4.22 on 2023-12-14T01:59:30
- Latest release: 0.6.1 on 2025-05-21T18:12:43
| Dependency | Optional |
|---|---|
| jax-cuda12-pjrt | false |
| nvidia-cublas-cu12 | true |
| nvidia-cuda-cupti-cu12 | true |
| nvidia-cuda-nvcc-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 |