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

[1bit_adam]
mpi4py

[all]
tqdm
recommonmark
pytest-forked
psutil
clang-format
sphinx
pre-commit
sphinx-rtd-theme
pytest
mpi4py

[dev]
pytest
pytest-forked
pre-commit
clang-format
sphinx
recommonmark
sphinx-rtd-theme

[readthedocs]
tqdm
psutil
