Training Neural Surrogates for Geophysical Fluid Dynamics: 2D Shallow Water Equations on PDEBench
Single-step validation loss is dangerously misleading when training neural operators on fluid PDEs. Here is what we learned building an end-to-end training pipeline for the 2D Shallow Water Equations on PDEBench, formatting for The Well, and evaluating autoregressive rollouts.