torch>=2.0
torch_ema
numpy
tqdm
scipy
