PINNs-JAX, Physics-informed Neural Networks (PINNs) implemented in JAX.
Primary LanguagePythonMIT LicenseMIT