Neural Operator Surrogate for Magnetized Plasma Simulations

We are releasing the code for our recent paper on Fourier Neural Operators (FNOs) applied to magnetohydrodynamic turbulence. The repository contains a PyTorch implementation of a FNO trained to act as a surrogate for 2D MHD simulations of the Orszag–Tang vortex, a canonical plasma physics benchmark. Given a short initial window of simulation frames plus two physical parameters (kinematic viscosity…