© 2025-2026 PySpect
| Dependency | Optional | Group |
|---|---|---|
| numpy | false | |
| packaging | false | |
| torch | false | |
| accelerate | true | training |
| Package | Optional | Group |
|---|---|---|
| megatron-bridge | false | |
| mbridge | true | megatron |
| av | true | dev |
| causal-conv1d | true | ssm |
| datasets | true | dev |
| einops | true | dev |
| emerging-optimizers | true | dev |
| fast-hadamard-transform | true | dev |
| fastapi | true | dev |
| flash-linear-attention | true | dev |
| flashinfer-python | true | dev |
| flask-restful | true | training |
| hypercorn | true | dev |
| mamba-ssm | true | ssm |
| megatron-energon | true | dev |
| multi-storage-client | true | dev |
| nvidia-modelopt | true | dev |
| nvidia-resiliency-ext | true | dev |
| omegaconf | true | training |
| onnxscript | true | dev |
| openai | true | dev |
| opentelemetry-api | true | dev |
| orjson | true | dev |
| quart | true | dev |
| sentencepiece | true | training |
| tensorstore | true | dev |
| tiktoken | true | training |
| tqdm | true | dev |
| transformer-engine | true | te |
| transformers | true | training |
| wandb | true | training |
| wget | true | dev |