torch
numpy
torchvision

[dev]
