torch>=2.4
triton>=3.0

[test]
pytest
