numpy
torch>=2.5.0
triton>=3.1.0
tqdm
