torch>=1.1.0
numba
scikit-learn
