torch>=1.2
torchvision>=0.4.0
tqdm
psutil
cpufeature
tensorboardX==1.8

[1bit_adam]
mpi4py

[all]
clang-format
sphinx-rtd-theme
pytest-forked
mpi4py
triton>=0.2.2
sphinx
recommonmark
pre-commit
pytest

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

[sparse_attn]
triton>=0.2.2
