torch
torchvision
PIL
timm
yaml
einops