torch
torchvision
scipy