numpy>=1.21.0
torch>=1.10.0
scikit-learn
