GitHub repository: yonashub/skorch
Description: A scikit-learn compatible neural network library that wraps pytorch