blobfile>=3.0.0
datasets<=2.21.0,>=2.20.0
packaging<26.0,>=23.0
torchdata<1.0,>=0.8.0
tiktoken>=0.9.0
psutil
timm
wandb
setuptools
einops>=0.8.1
matplotlib>=3.7

[dev]
pre-commit<5.0,>=4.0.0
ruff<1.0,>=0.7.0
pytest<8.0,>=6.0.0
expecttest<0.4,>=0.3.0

[gpu]
torch==2.11.0+cu130
torchvision==0.26.0+cu130
torchaudio==2.11.0+cu130
torchcodec==0.12.0
nvidia-cusparselt-cu13
nvidia-nccl-cu13
nvidia-cutlass-dsl>=4.4.2
nvidia-cudnn-frontend>=1.24.0
cuda-python==13.0.3
flash-attn-4==4.0.0b16
flash-qla==0.1.2
liger-kernel
flash-linear-attention
quack-kernels[cu13]==0.5.0
gram-newton-schulz==0.1.6
torch-c-dlpack-ext
apache-tvm-ffi==0.1.9
tile-kernels==1.0.0
tilelang==0.1.9
librosa<0.12,>=0.11.0
soundfile<0.14,>=0.13.1
diffusers==0.37.0
ftfy
peft==0.18.1
megatron-energon>=7.2.1
hf_transfer

[gpu:python_version == "3.11" and platform_machine == "aarch64"]
av<14.3,>=14.2.0

[gpu:python_version == "3.11" and platform_machine == "x86_64"]
av<15.0,>=14.3.0

[gpu:python_version == "3.12"]
av<14.3,>=14.0.0

[gpu:sys_platform == "linux" and (platform_machine == "aarch64" or platform_machine == "x86_64")]
flash-attn-3
flash-mla

[gpu:sys_platform == "linux" and platform_machine == "x86_64"]
flash-attn

[magi]
veomni[gpu]
debugpy

[magi:sys_platform == "linux"]
create-block-mask-cuda==0.1
flash-attn-cute==0.0.2
magi-to-hstu-cuda==0.1
magi-attention==1.1.1

[npu]
torch==2.10.0+cpu
torch-npu==2.10.0
torchvision==0.25.0+cpu
torchaudio==2.10.0+cpu
torchcodec==0.10.0
decorator>=5.2.1
scipy>=1.16.2
librosa<0.12,>=0.11.0
soundfile<0.14,>=0.13.1
diffusers==0.37.0
ftfy
peft==0.18.1
megatron-energon>=7.2.1
hf_transfer

[npu:python_version == "3.11"]
av<15.0,>=14.3.0

[npu:python_version == "3.12"]
av<14.3,>=14.0.0

[npu_aarch64]
torch==2.10.0
torch-npu==2.10.0
torchvision==0.25.0
torchaudio==2.10.0
decorator>=5.2.1
scipy>=1.16.2
librosa<0.12,>=0.11.0
soundfile<0.14,>=0.13.1
diffusers==0.37.0
ftfy
peft==0.18.1
megatron-energon>=7.2.1
hf_transfer

[npu_aarch64:python_version == "3.11"]
av<14.3,>=14.2.0

[npu_aarch64:python_version == "3.12"]
av<14.3,>=14.0.0
