torch>=1.9.0
numpy>=1.19.0

[dev]
pytest>=6.0
matplotlib>=3.3.0
black
isort
flake8

[test]
pytest>=6.0
matplotlib>=3.3.0
