© 2025-2026 PySpect
| Dependency | Optional | Group |
|---|---|---|
| jaxlib | false | |
| ml-dtypes | false | |
| numpy | false | |
| opt-einsum | false |
| scipy | false |
| jax-cuda12-plugin | true | cuda |
| jax-cuda13-plugin | true | cuda13 |
| jax-oneapi-plugin | true | oneapi |
| jax-rocm7-plugin | true | rocm7-local |
| kubernetes | true | k8s |
| libtpu | true | tpu |
| requests | true | tpu |
| xprof | true | xprof |