torch>=2.0

[dev]
pytest>=7.0
torchvision>=0.15
numpy>=1.20
matplotlib>=3.5

[viz]
numpy>=1.20
matplotlib>=3.5
