© 2025-2026 PySpect
| Dependency | Optional | Group |
|---|---|---|
| nvidia-cublas-cu12 | false | |
| nvidia-cusparse-cu12 | false | |
| nvidia-nvjitlink-cu12 | false |
| Package | Optional | Group |
|---|---|---|
| deepmd-kit | true | cu12 |
| gptqmodel | true | marlin-cuda12 |
| jax-cuda12-plugin | true | with-cuda |
| tensorflow | true | and-cuda |
| tensorflow-aarch64 | true | and-cuda |
| tensorflow-cpu | true | and-cuda |
| tensorflow-cpu-aws | true | and-cuda |
| tensorflow-intel | true | and-cuda |
| tensorflow-macos | true | and-cuda |
| tf-nightly | true | and-cuda |