I wrote a model that solves equations, it\'s based on a paper. Its uses the gradient of the network in the loss. By comparing the gradient to the model output, you can see i