scikit-learn
skorch
