transformers
accelerate
torch
einops