1
|
Pederson R, Kozlowski J, Song R, Beall J, Ganahl M, Hauru M, Lewis AGM, Yao Y, Mallick SB, Blum V, Vidal G. Large Scale Quantum Chemistry with Tensor Processing Units. J Chem Theory Comput 2023; 19:25-32. [PMID: 36508260 DOI: 10.1021/acs.jctc.2c00876] [Citation(s) in RCA: 6] [Impact Index Per Article: 6.0] [Reference Citation Analysis] [Abstract] [Track Full Text] [Journal Information] [Subscribe] [Scholar Register] [Indexed: 12/14/2022]
Abstract
We demonstrate the use of Googles cloud-based Tensor Processing Units (TPUs) to accelerate and scale up conventional (cubic-scaling) density functional theory (DFT) calculations. Utilizing 512 TPU cores, we accomplish the largest such DFT computation to date, with 247848 orbitals, corresponding to a cluster of 10327 water molecules with 103270 electrons, all treated explicitly. Our work thus paves the way toward accessible and systematic use of conventional DFT, free of any system-specific constraints, at unprecedented scales.
Collapse
Affiliation(s)
- Ryan Pederson
- Department of Physics and Astronomy, University of California, Irvine, California92617, United States.,X, the Moonshot Factory, Mountain View, California94043, United States.,Sandbox@Alphabet, Mountain View, California94043, United States
| | - John Kozlowski
- X, the Moonshot Factory, Mountain View, California94043, United States.,Sandbox@Alphabet, Mountain View, California94043, United States.,Department of Chemistry, University of California, Irvine, California92617, United States
| | - Ruyi Song
- X, the Moonshot Factory, Mountain View, California94043, United States.,Sandbox@Alphabet, Mountain View, California94043, United States.,Department of Chemistry, Duke University, Durham, North Carolina27708, United States
| | - Jackson Beall
- Sandbox@Alphabet, Mountain View, California94043, United States.,SandboxAQ, Palo Alto, California94304, United States
| | - Martin Ganahl
- Sandbox@Alphabet, Mountain View, California94043, United States.,SandboxAQ, Palo Alto, California94304, United States
| | - Markus Hauru
- Sandbox@Alphabet, Mountain View, California94043, United States.,The Alan Turing Institute, 96 Euston Road, LondonNW1 2DB, England, U.K
| | - Adam G M Lewis
- Sandbox@Alphabet, Mountain View, California94043, United States.,SandboxAQ, Palo Alto, California94304, United States
| | - Yi Yao
- Thomas Lord Department of Mechanical Engineering and Materials Science, Duke University, Durham, North Carolina27708, United States
| | - Shrestha Basu Mallick
- X, the Moonshot Factory, Mountain View, California94043, United States.,Sandbox@Alphabet, Mountain View, California94043, United States
| | - Volker Blum
- Department of Chemistry, Duke University, Durham, North Carolina27708, United States.,Thomas Lord Department of Mechanical Engineering and Materials Science, Duke University, Durham, North Carolina27708, United States
| | - Guifre Vidal
- X, the Moonshot Factory, Mountain View, California94043, United States.,Sandbox@Alphabet, Mountain View, California94043, United States.,Google Quantum AI, Mountain View, California94043, United States
| |
Collapse
|