setuptools>=52.0.0
torch>=1.7.1
keras>=2.4.3
numpy
scipy
tqdm
h5py
