torch<2.0.0,>=1.9.0
torchvision<1.0.0,>=0.7.0
einops<1.0.0,>=0.3.0

[docs]
matplotlib
sphinx-gallery
sphinx-rtd-theme
memory_profiler
tabulate

[lint]
darglint
flake8
mccabe
pycodestyle
pydocstyle
pyflakes
pep8-naming
flake8-bugbear
flake8-comprehensions
flake8-tidy-imports
black
isort

[test]
scipy
pytest<5.0.0,>=4.5.0
pytest-benchmark<4.0.0,>=3.2.2
pytest-optional-tests>=0.1.1
pytest-cov
coveralls
