/Physics-informed-neural-network-in-JAX

Example problems in Physics informed neural network in JAX

Primary LanguageJupyter NotebookMIT LicenseMIT

Example notebooks for various applications in Physics informed neural network.

Description Functional form Class form ✨New
[ODE] Open In Colab Open In Colab
[PDE] Burgers Open In Colab
[PDE] Laplace Open In Colab Open In Colab
[ODE] Supervised loss + PINN Open In Colab
[ODE] System of ODE Open In Colab
[ODE] Finite difference Open In Colab

If you find it useful please give it a ⭐