Model-parallel Fourier neural operators as learned surrogates for large-scale parametric PDEs

Published: 01 Jan 2023, Last Modified: 16 Aug 2024Comput. Geosci. 2023EveryoneRevisionsBibTeXCC BY-SA 4.0
Abstract: Highlights•Novel approach to parallelism in Fourier neural operators using domain decomposition.•Mathematical description of parallelism in the context of operator learning.•Distributed training of a learned surrogate for subsurface two-phase flow.•Largest forward inference and gradient computation of a Fourier neural operator.
Loading