© 2025-2026 PySpect
| Dependency | Optional | Group |
|---|---|---|
| jax-cuda13-pjrt | false | |
| nvidia-cublas | true | with-cuda |
| nvidia-cuda-cupti | true | with-cuda |
| nvidia-cuda-nvcc | true | with-cuda |
| Package | Optional | Group |
|---|---|---|
| jax | true | cuda13 |
| nvidia-cuda-nvrtc | true | with-cuda |
| nvidia-cuda-runtime | true | with-cuda |
| nvidia-cudnn-cu13 | true | with-cuda |
| nvidia-cufft | true | with-cuda |
| nvidia-cusolver | true | with-cuda |
| nvidia-cusparse | true | with-cuda |
| nvidia-nccl-cu13 | true | with-cuda |
| nvidia-nvjitlink | true | with-cuda |
| nvidia-nvshmem-cu13 | true | with-cuda |
| nvidia-nvvm | true | with-cuda |