transformers
torch
einops
einops-exts