numpy>1.20.0,<2.0.0
torch>=1.10.0
scikit-learn>=1.0.0
