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