click
numpy
Pillow
torch>=1.7.0
torchvision

[docs]
sphinx-copybutton>=0.4.0
sphinx-rtd-theme>=1.0.0
sphinxcontrib.datatemplates>=0.9.0
sphinxcontrib.bibtex>=2.4.1
nbsphinx>=0.8.8
ipykernel>=6.13.0

[tests]
pytest
pytest-cov
