torch>=1.1.0
tqdm
scikit-learn
tensorflow
