accelerate
timm
sentencepiece
einops
torch
torchvision
transformers