transformers<5.9.0,>=5.8.1
mistral-common>=1.10.0
peft>=0.18.1
datasets>=2.20.0
accelerate
diffusers>=0.36.0
peft>=0.18.0
einops
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
nvidia-resiliency-ext
flash-linear-attention
timm
open-clip-torch>=3.2.0
mlflow>=3.9.0
comet-ml>=3.50.0
torch>=2.6.0
flashinfer-python==0.6.8.post1
flashinfer-cubin==0.6.8.post1

[audio]
librosa

[parquet]
pyarrow>=14.0.0

[recipes]
nemo-run

[ssm]
mamba-ssm
causal-conv1d

[te]
transformer-engine[core_cu13,pytorch]

[tensor-inspect]
nvdlfw-inspect==0.2.1
