Add 2D wave equation FNO training recipe#1489

Open

gpartin wants to merge 2 commits into

NVIDIA:mainfrom

gpartin:feature/add-wave-equation-example

Commits on Mar 14, 2026

Add 2D wave equation FNO training recipe
gpartin
committed
Address review: remove dead code, fix device defaults, add input validation, reuse loss_fun, add seed/reproducibility support, fix f-string, add vectorization comment
gpartin
committed