monai
albumentations
torch
numpy
torchvision