torch
monai
torchvision