torch>=1.9.0
matplotlib>=3.3.0

[dev]
pytest
pytest-cov
