torch
torchvision
monai