torch>=1.5.1
numpy
scikit-learn
matplotlib
termcolor==1.1.0
pyzmq>=19.0.0
flatbuffers==1.12
pydotplus==2.0.2
pyyaml>=3.13

[dev]
pytest
pytest-cov
pytest-xdist
docker
sphinx
sphinx_rtd_theme
nbsphinx
sphinxcontrib-bibtex
