torch
transformers
# flash-attn
einops