Large Scale Quantum Chemistry with Tensor Processing Units.
Ryan D PedersonJohn KozlowskiRuyi SongJackson BeallMartin GanahlMarkus HauruAdam G M LewisYi YaoShrestha Basu MallickVolker BlumGuifre VidalPublished in: Journal of chemical theory and computation (2022)
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.