numpy>=1.18.0
pytorch-lightning
scikit-image>=0.16.2
scikit-learn>=0.23.2
tensorly>=0.5.1
torch>=1.7.0
torchvision
scipy>=1.5.4

[dev]
ipykernel
ipython
matplotlib
torchsummary>=1.5.0
yacs>=0.1.7
black
coverage
flake8
flake8-print
isort
m2r
mypy
nbsphinx
nbval
pre-commit
pytest
pytest-cov
sphinx
sphinx-rtd-theme

[extras]
ipykernel
ipython
matplotlib
torchsummary>=1.5.0
yacs>=0.1.7
