torchvision
timm
einops
ftfy