Towards Digital Twins for NVIDIA’s Earth-2 Initiative: Pushing the Limits of Deep Auto-regressive Fourier Neural Operator and Transformer models for Earth System Emulation

Kashinath, Karthik

Towards Digital Twins for NVIDIA’s Earth-2 Initiative: Pushing the Limits of Deep Auto-regressive Fourier Neural Operator and Transformer models for Earth System Emulation

Karthik Kashinath¹, Mike S Pritchard¹, Anima Anandkumar¹, Jaideep Pathak¹, Morteza Mardani¹, Thorsten Kurth², David Hall³, Peter Messmer¹, Stan Posey¹, Shashank Subramanian⁴, Peter Harrington⁴ and NVIDIA Earth-2 and collaborators, (1)NVIDIA Corporation, Santa Clara, United States, (2)NVIDIA Corporation, Zurich, Switzerland, (3)NVIDIA Corporation, Lafayette, United States, (4)Lawrence Berkeley National Laboratory, NERSC, Berkeley, United States