AI Agent Research Hub

Mar 14, 2026 · Artificial Intelligence

Adaptive-Weight NTK-PINN Solves High-Frequency Wave Equation Using JAX

This tutorial explains how the Neural Tangent Kernel (NTK) perspective reveals the loss‑balance problem in Physics‑Informed Neural Networks (PINNs), introduces an NTK‑based adaptive‑weight algorithm, provides a full JAX implementation for a 1‑D high‑frequency wave equation, and shows that input normalisation dramatically improves accuracy while only modestly increasing training time.