This repository contains the following notebooks about Physics Informed Neural Networks (PINNs):
📌 Notebook #1: PINN for solving Burgers' Equation, written in pure JAX and in JAX + Flax
📌 Notebook #2: JAX + Flax PINNs for 1D and 2D Diffusion Equations
📌 Notebook #3: JAX + Flax PINNs for Coupled PDEs with Periodic BCs
📌 Notebook #4: JAX + Flax PINNs for unknown parameter estimation