einops>=0.4
torch>=1.6
triton>=2.0dev
