© 2025-2026 PySpect
| Dependency | Optional | Group |
|---|---|---|
| blobfile | false | |
| datasets | false | |
| einops | false | |
| matplotlib | false | |
| packaging | false | |
| psutil | false | |
| setuptools | false | |
| tiktoken | false | |
| timm | false | |
| torchdata | false | |
| wandb | false | |
| apache-tvm-ffi | true | gpu |
| av | true | gpu |
| create-block-mask-cuda | true | magi |
| cuda-python | true | gpu |
| debugpy | true | magi |
| decorator | true | npu |
| diffusers | true | gpu |
| expecttest | true | dev |
| flash-attn | true | gpu |
| flash-attn-3 | true | gpu |
| flash-attn-4 | true | gpu |
| flash-attn-cute | true | magi |
| flash-linear-attention | true | gpu |
| flash-mla | true | gpu |
| flash-qla | true | gpu |
| ftfy | true | gpu |
| gram-newton-schulz | true | gpu |
| hf_transfer | true | gpu |
| librosa | true | gpu |
| liger-kernel | true | gpu |
| magi-attention | true | magi |
| magi-to-hstu-cuda | true | magi |
| megatron-energon | true | gpu |
| nvidia-cudnn-frontend | true | gpu |
| nvidia-cusparselt-cu13 | true | gpu |
| nvidia-cutlass-dsl | true | gpu |
| nvidia-nccl-cu13 | true | gpu |
| peft | true | gpu |
| pre-commit | true | dev |
| pytest | true | dev |
| quack-kernels | true | gpu |
| ruff | true | dev |
| scipy | true | npu |
| soundfile | true | gpu |
| tile-kernels | true | gpu |
| tilelang | true | gpu |
| torch | true | gpu |
| torch-c-dlpack-ext | true | gpu |
| torch-npu | true | npu |
| torchaudio | true | gpu |
| torchcodec | true | gpu |
| torchvision | true | gpu |