torch>=2.6.0
numpy
packaging>=24.2

[dev]
transformer-engine[core_cu13,pytorch]
nvidia-resiliency-ext
tqdm
einops~=0.8
tensorstore!=0.1.46,!=0.1.72,~=0.1
nvtx~=0.2
multi-storage-client~=0.27
opentelemetry-api~=1.33.1
mamba-ssm~=2.2
causal-conv1d~=1.5
flash-linear-attention~=0.4.0
megatron-energon[av_decode]~=6.0
av
flashinfer-python~=0.5.0
wget
onnxscript
fastapi~=0.50
datasets
emerging_optimizers
hypercorn
quart
openai[aiohttp]
orjson

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

[lts]
tqdm
einops~=0.8
tensorstore!=0.1.46,!=0.1.72,~=0.1
nvtx~=0.2
multi-storage-client~=0.27
opentelemetry-api~=1.33.1
mamba-ssm~=2.2
causal-conv1d~=1.5
megatron-energon[av_decode]~=6.0
av
flashinfer-python~=0.5.0
wget
onnxscript
fastapi~=0.50
datasets
emerging_optimizers

[mlm]
flask-restful
sentencepiece
tiktoken
wandb
transformers
accelerate

[training]
flask-restful
sentencepiece
tiktoken
wandb
transformers
accelerate
