transformers
timm
einops
accelerate