numpy
torch>=2.5.0
triton>=3.3.0
tqdm
