torch>=2.0
triton>=2.0
numpy
tqdm
