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