2Program in Applied and Computational Mathematics, Princeton University, Princeton, New Jersey 08544, USA
3Department of Materials Science and Engineering, Stanford University, Stanford, California 94305, USA
4Department of Chemistry, Department of Physics, Princeton Materials Institute, Princeton University, Princeton, New Jersey 08544, USA
*Present address: AI for Science Institute, Beijing, China and Center for Machine Learning Research and School of Mathematical Sciences, Peking University, Beijing, China.
