torch
tqdm
ninja
numpy
psutil
packaging
hjson
triton

[1bit]

[1bit_mpi]
mpi4py

[all]
hjson
recommonmark
megatron-lm==1.1.5
torch
importlib-metadata>=4
clang-format
docutils<0.18
pytest
mpi4py
tqdm
pytest-forked
pre-commit
xgboost
psutil
pytest-randomly
sphinx-rtd-theme
sphinx
tabulate

[autotuning]
tabulate

[autotuning_ml]
hjson
xgboost
tabulate

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

[readthedocs]
tqdm
psutil
docutils<0.18
hjson
torch
