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

[dev]
pytest
pytest-cov
pytest-xdist
docker

[docs]
sphinx==3.2.1
sphinx_rtd_theme==0.5.2
jupyter-sphinx==0.3.2
myst-nb==0.12.3
