Repository navigation

FNO: run spectral convolutions in float32 under autocast - #2150

Open

GeorgeLiang3 wants to merge 2 commits into

NVIDIA:mainfrom

GeorgeLiang3:fno-amp-fp32

Open

FNO: run spectral convolutions in float32 under autocast#2150
GeorgeLiang3 wants to merge 2 commits into
NVIDIA:mainfrom
GeorgeLiang3:fno-amp-fp32

Commits on Oct 30, 2026
- fix(fno): run spectral convolutions in float32 under autocast
  GeorgeLiang3
  committed
  05ae2ed
- test(fno): check gradients and weight updates in the autocast training test
  GeorgeLiang3
  committed
  92f5082