numpy>=1.18.0
torch>=1.7.0
torchvision>=0.8.1
tensorly

[dev]
black
twine
pre-commit

[docs]
ipython
ipykernel
sphinx
sphinx_rtd_theme
nbsphinx
m2r

[pipeline]
pytorch-lightning

[test]
flake8
flake8-print
pytest
nbval

[utils]
matplotlib
tqdm
torchsummary>=1.5.0
yacs>=0.1.7
