scikit-learn
torch