torch>=2.2.1
triton>=2.2.0
