torch>=1.9.0

[dev]
pytest
black
flake8
numpy
matplotlib

[examples]
numpy
matplotlib
