Physics Informed Learning, Neural Networks, Machine Learning, Differential Equations, Math.
to create a neural network architecture: import torch import torch.nn as nn class NeuralNet(nn.Module): def __init__(self, hidden_size, output_size=1,input_size=1): super(NeuralNet, self).__init__() self.l1 = nn.Linear(input_size, hidden_size) self.relu1 = nn.LeakyReLU() self.l2 =