ninja
numpy
packaging
psutil
tensorboardX==1.8
torch>=1.2
torchvision>=0.4.0
tqdm
triton

[1bit_adam]
mpi4py

[all]
clang-format
importlib-metadata>=4
megatron-lm==1.1.5
mpi4py
pre-commit
psutil
pytest
pytest-forked
pytest-randomly
recommonmark
sphinx
sphinx-rtd-theme
tqdm

[dev]
clang-format
importlib-metadata>=4
megatron-lm==1.1.5
pre-commit
pytest
pytest-forked
pytest-randomly
recommonmark
sphinx
sphinx-rtd-theme

[readthedocs]
psutil
tqdm
