einops
torch>=2.6.0
triton>=3.2.0
packaging
pytest
