torch>=2.2
tqdm
numpy
