torch
torchvision
numpy

[dev]
scipy
matplotlib
jupyter
pytest
