datasets>=4.0.0
diffusers
ftfy
imageio-ffmpeg
megatron-fsdp
opencv-python-headless==4.10.0.84
pybind11
pyyaml
torch<=2.9.0
torchdata
transformers<=4.57.1
wandb
torchao
mlflow

[:platform_machine == "x86_64" and platform_system != "Darwin"]
liger-kernel>=0.5.9

[all]
nemo_automodel[extra]
nemo_automodel[vlm]

[extra]
flash-linear-attention
perceptron
sentencepiece

[fa]
flash-attn<=2.8.3

[moe]
transformer-engine[pytorch]==2.8.0

[vlm]
pillow
qwen-omni-utils
timm==1.0.16
backoff
numpy
numba
mistral_common[opencv]

[vlm:platform_machine == "x86_64" and platform_system != "Darwin"]
qwen-vl-utils[decord]
torchcodec
