torch
torchvision
einops
Pillow
requests