GitHub repository: skorch-dev/skorch
Primary language: Jupyter Notebook
Description: A scikit-learn compatible neural network library that wraps PyTorch