Large scale quantum chemistry with Tensor Processing Units
Abstract
We demonstrate the use of Google's 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 towards accessible and systematic use of conventional DFT, free of any system-specific constraints, at unprecedented scales.
- Publication:
-
arXiv e-prints
- Pub Date:
- February 2022
- DOI:
- 10.48550/arXiv.2202.01255
- arXiv:
- arXiv:2202.01255
- Bibcode:
- 2022arXiv220201255P
- Keywords:
-
- Physics - Computational Physics;
- Physics - Chemical Physics
- E-Print:
- doi:10.1021/acs.jctc.2c00876