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

[1bit_adam]
mpi4py
cupy-cuda100

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

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

[readthedocs]
tqdm
psutil
