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