torch>=2.0.0
torchvision>=0.15.0
numpy>=1.20.0
tqdm>=4.60.0

[dev]
pytest>=7.0.0
black>=22.0.0
flake8>=4.0.0

[viz]
matplotlib>=3.5.0
