skorch “A scikit learn compatible neural network library that wraps PyTorch“ https://skorch.readthedocs.io/en/stable/index.html