1. Program in Applied and Computational Mathematics, Princeton University, Princeton, NJ 08544
2. Department of Chemistry and Princeton Materials Institute, Princeton University, Princeton, NJ 08544
3. Department of Physics, Princeton University, Princeton, NJ 08544
4. AI for Science Institute, Beijing 100080, China
5. Center for Machine Learning Research and School of Mathematical Sciences, Peking University, Beijing 100084, China