torch>=1.13
torchvision
matplotlib
seaborn
tqdm

[dev]
flake8
mypy==0.961
black==22.3.0
isort==5.10.1
pytest
pytest-sphinx
pytest-cov
twine>=1.11.0
setuptools
wheel
Sphinx<5.1.0,>=4.3.0
furo==2022.6.4.1
myst-parser<0.19.0,>=0.15.2
sphinx-copybutton==0.5.0
sphinx-autobuild==2021.3.14
sphinx-autodoc-typehints
packaging
