torch>=2.0.0
einops>=0.6.1

[triton]
triton>=2.0.0
