datasets>=4.0.0
diffusers
ftfy
imageio-ffmpeg
megatron-fsdp>=0.2.3
opencv-python-headless==4.10.0.84
pybind11
pyyaml
torch<=2.10.0,>=2.6.0
torchdata
transformers>=5.0.0
wandb
torchao
mlflow

[all]
nemo_automodel[cuda]
nemo_automodel[delta-databricks]
nemo_automodel[extra]
nemo_automodel[vlm]

[cuda]
bitsandbytes
causal-conv1d
mamba-ssm
nv-grouped-gemm
onnxscript>=0.5.6
transformer-engine[pytorch]<=2.11.0

[delta-databricks]
deltalake>=1.0.0
databricks-sql-connector>=3.0.0

[extra]
flash-linear-attention
perceptron
sentencepiece

[fa]
flash-attn<=2.8.3

[moe]
nemo_automodel[cuda]
deep_ep

[vlm]
albumentations
backoff
mistral_common[opencv]>=1.9.0
numpy
numba
open-clip-torch
pillow
qwen-omni-utils
timm<=1.0.22

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