accelerate
timm
einops
torch
torchvision
transformers