torch
tqdm
tensorboardX==1.8
ninja
numpy
psutil
packaging
triton

[1bit]

[1bit_mpi]
mpi4py

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

[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
