torch>=1.1.0
numba
numpy<1.20
scikit-learn
