scikit-learn>=0.16
numba>=0.34
torch>=1.0
