einops
torch>=1.6
triton>=1.0.1dev
