torch
torchvision
pillow
timm
pyyaml
einops