einops
flask-restful
nltk
pytest
pytest-cov
pytest_mock
pytest-random-order
sentencepiece
tiktoken
wrapt
zarr
wandb
tensorstore!=0.1.46,!=0.1.72
torch
torch
packaging

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