feat(skorch): add an inherited class from skorch.NeuralNet that is compatible with PyTorch Frame

feat(skorch): add an inherited class from skorch.NeuralNet that is compatible with PyTorch Frame #2951