accelerate
einops
transformers
torch