torch
torchvision
transformers
einops
matplotlib
timm