torch>=1.9

[build]
setuptools
wheel
twine

[dev]
black
flake8
flake8-black
isort>=5.7
pytest
pytest-cov
ptflops>=0.6.8
numpy
pytorch_lightning
onnxruntime~=1.13

[docs]
sphinx==5.0.0
sphinxcontrib.katex==0.8.6
matplotlib==3.5.3
tensorboard==2.10.0
python-etcd==0.4.5
sphinx-copybutton==0.5.0
sphinx-panels==0.4.1
myst-parser==0.18.1
