einops
torch>=1.6
triton==1.0.1.dev20210924
