torch>=1.7
numpy>=1.18
opt_einsum
scipy>=0.16
ninja

[dev]
pytest>=6.0
sphinx
sphinx_rtd_theme
ipykernel
matplotlib
nbsphinx
torchvision
