torch
torchvision
timm
numpy
pillow
transformers
einops