torch>=2.8
triton>=3.4
