numpy
torch>=2.6.0
tqdm
