torch==1.7.0
numpy
pandas
scikit-learn
numba
