torch>=1.9.0
matplotlib>=3.4.0
numpy>=1.20.0

[dev]
pytest>=7.0
torchvision>=0.10.0
