numpy
tqdm
scikit-learn
torch<2.0
