accelerate
timm
einops
torch
torchvision
Pillow
transformers
sentencepiece