torch>=1.10
numpy>=1.20

[dev]
pytest>=7.0
matplotlib>=3.5

[viz]
matplotlib>=3.5
