torch
numpy
torchvision
matplotlib

[dev]
pytest
black
flake8
isort
