numpy>=1.18.0
pytorch-lightning
scikit-image
scikit-learn
tensorly
torch>=1.7.0
torchvision

[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
