pinnx 0.0.1

Physics-informed.

PINNx: Physics-Informed Neural Networks for Scientific Machine Learning in JAX

PINNx is a library for scientific machine learning and physics-informed learning. It is rewritten according to DeepXDE but is enhanced by our Brain Dynamics Programming (BDP) ecosystem. For example, it leverages brainstate for just-in-time compilation, brainunit for dimensional analysis, braintools for checkpointing, loss functions, and other utilities.

PINNx implements the following algorithms, but with the flexibility and efficiency of JAX:

• solving different PINN problems
  • solving forward/inverse ordinary/partial differential equations (ODEs/PDEs) [SIAM Rev.]
  • solving forward/inverse integro-differential equations (IDEs) [SIAM Rev.]
  • fPINN: solving forward/inverse fractional PDEs (fPDEs) [SIAM J. Sci. Comput.]
  • NN-arbitrary polynomial chaos (NN-aPC): solving forward/inverse stochastic PDEs (sPDEs) [J. Comput. Phys.]
  • PINN with hard constraints (hPINN): solving inverse design/topology optimization [SIAM J. Sci. Comput.]
• improving PINN accuracy
  • residual-based adaptive sampling [SIAM Rev., Comput. Methods Appl. Mech. Eng.]
  • gradient-enhanced PINN (gPINN) [Comput. Methods Appl. Mech. Eng.]
  • PINN with multi-scale Fourier features [Comput. Methods Appl. Mech. Eng.]
• (physics-informed) deep operator network (DeepONet)
  • DeepONet: learning operators [Nat. Mach. Intell.]
  • DeepONet extensions, e.g., POD-DeepONet [Comput. Methods Appl. Mech. Eng.]
  • MIONet: learning multiple-input operators [SIAM J. Sci. Comput.]
  • Fourier-DeepONet [Comput. Methods Appl. Mech. Eng.], Fourier-MIONet [arXiv]
  • physics-informed DeepONet [Sci. Adv.]
  • multifidelity DeepONet [Phys. Rev. Research]
  • DeepM&Mnet: solving multiphysics and multiscale problems [J. Comput. Phys., J. Comput. Phys.]
  • Reliable extrapolation [Comput. Methods Appl. Mech. Eng.]
• multifidelity neural network (MFNN)
  • learning from multifidelity data [J. Comput. Phys., PNAS]

Installation

• Install the stable version with pip:

pip install pinnx --upgrade

Documentation

The official documentation is hosted on Read the Docs: https://pinnx.readthedocs.io/

See also the BDP ecosystem

We are building the Brain Dynamics Programming ecosystem: https://ecosystem-for-brain-dynamics.readthedocs.io/