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
megatron-core
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

[:sys_platform != "darwin"]
nvidia-modelopt[torch]
nvidia-resiliency-ext

[recipes]
nemo-run

[te]

[te:sys_platform != "darwin"]
transformer-engine[pytorch]==2.4.0
