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

[audio]
torchcodec
av<15.0,>=14.3.0
librosa<0.12,>=0.11.0
soundfile<0.14,>=0.13.1

[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

[dit]
diffusers==0.36.0
torchcodec
av<15.0,>=14.3.0
librosa<0.12,>=0.11.0
soundfile<0.14,>=0.13.1
ftfy
peft

[fa4]
flash-attn-4
nvidia-cutlass-dsl>=4.4.2

[gpu]
torch==2.9.1+cu129
torchvision==0.24.1+cu129
torchaudio==2.9.1+cu129
nvidia-cusparselt-cu12
nvidia-nccl-cu12
liger-kernel
flash-attn
flash-attn-3
hf_transfer
torchcodec==0.8.1
cuda-python==12.9.1
torch-c-dlpack-ext
flash-linear-attention
quack-kernels==0.3.4

[megatron]
megatron-energon>=7.2.1

[npu]
torch==2.7.1+cpu
torch-npu==2.7.1
torchvision==0.22.1+cpu
torchaudio==2.7.1+cpu
decorator>=5.2.1
scipy>=1.16.2
torchcodec<0.6.0,>=0.4.0

[npu_aarch64]
torch==2.7.1
torch-npu==2.7.1
torchvision==0.22.1
torchaudio==2.7.1
decorator>=5.2.1
scipy>=1.16.2

[transformers5-exp]
transformers==5.2.0

[trl]
trl<=0.9.6

[video]
torchcodec
av<15.0,>=14.3.0
librosa<0.12,>=0.11.0
soundfile<0.14,>=0.13.1
