FudanEMWLab / PINN-1

Simple PyTorch Implementation of Physics Informed Neural Network (PINN)

Simple PyTorch Implementation of Physics Informed Neural Network (PINN)

This repository contains my simple and clear to understand implementation of the paper [1]. As an example, I solved the 1D heat partial differential equation.

3D plot of the solution to 1D Heat Equation.

All the steps are included in a single file: solve_PDE_NN.ipynb which depends on the pytorch and numpy packages.

[1] Raissi, Maziar, Paris Perdikaris, and George E. Karniadakis. "Physics-informed neural networks: A deep learning framework for solving forward and inverse problems involving nonlinear partial differential equations." Journal of Computational Physics 378 (2019): 686-707.