einops
timm
torch
torchvision
