transformers<5.0.0
datasets>=2.20.0
accelerate
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]<0.17.0,>=0.15.0a0
qwen-vl-utils
transformer-engine[core_cu13,pytorch]<2.13.0,>=2.10.0a0
mamba-ssm
nvidia-resiliency-ext~=0.4.1
causal-conv1d
flash-linear-attention
timm
open-clip-torch>=3.2.0
mlflow>=3.5.0
torch>=2.6.0

[recipes]
nemo-run

[tensor-inspect]
nvdlfw-inspect==0.2.1
