torch>=2.4
triton>=3.0
einops

[dev]
pytest
