torch>=2.1.0
scipy
einops
numpy
tqdm
