einops
flash-attn
torch
transformers