torch>=1.8
scikit-learn
pandas
