setuptools
torch
einops
flash_attn
transformers
