torch>=2.0
scipy
torch_ema
h5py
numpy
tqdm
