torch
tqdm
ninja
numpy
psutil
packaging
hjson
triton

[1bit]

[1bit_mpi]
mpi4py

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

[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
