torch
torchvision
torchmetrics
pillow
