transformers<=5.3.0,>=5.0.0
peft>=0.18.1
datasets>=2.20.0
accelerate
diffusers>=0.36.0
peft>=0.18.0
einops
imageio
imageio-ffmpeg
omegaconf>=2.3.0
tensorboard>=2.19.0
typing-extensions
rich
wandb>=0.25.0
six>=1.17.0
regex>=2024.11.6
pyyaml>=6.0.2
tqdm>=4.67.1
hydra-core<=1.3.2,>1.3
megatron-core[dev,mlm]
qwen-vl-utils
transformer-engine[core_cu13,pytorch]
mamba-ssm
nvidia-resiliency-ext
causal-conv1d
flash-linear-attention
timm
open-clip-torch>=3.2.0
mlflow>=3.9.0
comet-ml>=3.50.0
torch>=2.6.0

[parquet]
pyarrow>=14.0.0

[recipes]
nemo-run

[tensor-inspect]
nvdlfw-inspect==0.2.1
