timm
transformers
einops
accelerate
torch