Pretrained Model Checkpoints for "Forward and Inverse Mantle Convection with Neural Operators"

This record provides pretrained neural operator checkpoint files associated with the paper: Kong, C., Gurnis, M., and Ross, Z. E. (2026). Forward and Inverse Mantle Convection with Neural Operators. Geophysical Journal International. Accepted.

The checkpoint files support the code repository accompanying the paper and are intended for model loading, reproduction of example notebooks, including forward, reverse, Stokes physics-informed, and inverse mantle-convection demonstrations. The archive includes pretrained checkpoints for the forward convection neural operator, variable Rayleigh number forward convection neural operator, reverse convection neural operator, and Stokes physics-informed neural operator.