datasets>=4.0.0
megatron-fsdp==0.5.0
mistral-common[audio,hf-hub,sentencepiece]
ninja
pyyaml
tiktoken
torch>=2.8.0
torchdata
transformers==5.18.0
torchao>=0.16.0
flashoptim>=0.1.3

[:sys_platform == "linux"]
quack-kernels==0.6.4

[all]
nemo-automodel[cli]
nemo-automodel[cuda]
nemo-automodel[delta-databricks]
nemo-automodel[diffusion]
nemo-automodel[extra]
nemo-automodel[fla]
nemo-automodel[s3]
nemo-automodel[tracking]
nemo-automodel[vlm]

[cli]
pyyaml
nemo-run

[cuda]
causal-conv1d
mamba-ssm
nvidia-cudnn-frontend[cutedsl]>=1.27.0
onnxscript>=0.5.6
transformer-engine[pytorch]>=2.19.0
tilelang>=0.1.11
tile-kernels==1.0.0
apache-tvm-ffi<=0.1.11

[cuda_source]
bitsandbytes

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

[diffusion]
diffusers>=0.39.0
ftfy
imageio
torchvision

[diffusion-media]
av
imageio-ffmpeg
opencv-python-headless==4.10.0.84

[diffusion_kernels]
nemo-automodel[diffusion]
kernels

[extra]
perceptron
sentencepiece

[fa]
flash-attn<=2.8.3

[ffpa]
ffpa-attn>=0.2.4
nvidia-cutlass-dsl==4.6.2

[fla]
flash-linear-attention>=0.4.2

[media]
nemo-automodel[vlm-media]
nemo-automodel[diffusion-media]

[mlflow]
mlflow-skinny>=3.16.1
sqlalchemy<3,>=1.4.0
alembic!=1.10.0,<2

[mlflow-full]
mlflow>=3.16.1

[moe]
nemo-automodel[cuda]
nemo-automodel[fla]
deep_ep

[mok]

[mok:sys_platform == "linux" and python_version >= "3.12"]
mixture-of-kittens

[msa]

[msa:sys_platform == "linux"]
fmha-sm100
nvidia-cutlass-dsl==4.6.2

[msc]
multi-storage-client>=0.13

[s3]
boto3>=1.34

[tracking]
nemo-automodel[mlflow]
nemo-automodel[wandb]

[vlm]
backoff
mistral_common>=1.11.0
numpy
numba
open-clip-torch
pillow
soundfile
timm<=1.0.22

[vlm-media]
albumentations
opencv-python-headless==4.10.0.84
qwen-omni-utils
qwen-vl-utils

[vlm-media:platform_machine == "x86_64" and platform_system != "Darwin"]
decord

[wandb]
wandb>=0.30.0
