lululxvi / deepxde

A library for scientific machine learning and physics-informed learning
https://deepxde.readthedocs.io
GNU Lesser General Public License v2.1
2.76k stars 758 forks source link

Backend JAX supports 1D diffusion equation in PINN forward #1593

Closed ZongrenZou closed 11 months ago

ZongrenZou commented 11 months ago

Now both the forward and reverse modes support 1D diffusion equation in PINN forward.