torch>=1.9

[build]
setuptools
wheel
twine

[dev]
black==23.1.0
flake8==6.0.0
flake8-black==0.3.6
isort>=5.7
pytest
pytest-cov
ptflops==0.6.8
numpy
pytorch_lightning
onnx~=1.12
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
