accelerate>=1.6.0
datasets
numpy<2
omegaconf>=2.3.0
packaging
tensorboard>=2.19.0
torch
transformers>=4.51.3
typing-extensions
rich
wandb>=0.19.10
six>=1.17.0
regex>=2024.11.6
pyyaml>=6.0.2
einops>=0.8.1
sentencepiece>=0.2.0
tiktoken>=0.9.0
tqdm>=4.67.1
hydra-core<=1.3.2,>1.3
megatron-core<0.15.0,>=0.14.0a0

[:sys_platform != "darwin"]
nvidia-modelopt[torch]<0.34.0,>=0.33.0a0
nvidia-resiliency-ext<0.5.0,>=0.4.0a0
transformer-engine[pytorch]<2.8.0,>=2.6.0a0

[recipes]
nemo-run<0.6.0,>=0.5.0a0
