1. Program in Applied and Computational Mathematics, Princeton University, Princeton, New Jersey 08544, United States
2. AI for Science Institute, Beijing 100080, People’s Republic of China
3. DP Technology, Beijing 100080, People’s Republic of China
4. Center for Machine Learning Research, School of Mathematical Sciences, Peking University, Beijing 100084, People’s Republic of China
5. Department of Chemistry, Department of Physics, Princeton Institute for the Science and Technology of Materials, Princeton University, Princeton, New Jersey 08544, United States