torch
torchvision
numpy
einops
urllib