Physics Informed Learning, Neural Networks, Machine Learning, Differential Equations, Math.
Backward pass and weight update optimizer.zero_grad() loss.backward() optimizer.step() Notice the use of torch.autograd.grad function to automatically differentiate the output y_pred with respect to the input t to compute the loss function. Results After training, we can see that the loss function