The code was authored by the following people: Jaideep Pathak - NVIDIA Corporation Shashank Subramanian - NERSC, Lawrence Berkeley National Laboratory Peter Harrington - NERSC, Lawrence Berkeley National Laboratory Sanjeev Raja - NERSC, Lawrence Berkeley National Laboratory Ashesh Chattopadhyay - Rice University Morteza Mardani - NVIDIA Corporation Thorsten Kurth - NVIDIA Corporation David Hall - NVIDIA Corporation Zongyi Li - California Institute of Technology, NVIDIA Corporation Kamyar Azizzadenesheli - Purdue University Pedram Hassanzadeh - Rice University Karthik Kashinath - NVIDIA Corporation Animashree Anandkumar - California Institute of Technology, NVIDIA Corporation