torch
numpy<2.0.0

[dev]
tqdm
einops
tensorstore!=0.1.46,!=0.1.72
nvtx
transformers
multi-storage-client
setuptools<80.0.0
megatron-energon[av_decode]<7

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

[lts]
tqdm
einops
tensorstore!=0.1.46,!=0.1.72
nvtx
transformers
zarr
setuptools<80.0.0

[mlm]
flask-restful
sentencepiece
tiktoken
wandb
