Large scale quantum chemistry with Tensor Processing Units
arXiv:2202.01255 · doi:10.1021/acs.jctc.2c00876
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.
References in corpus (18)
- The SIESTA method for ab initio order-N materials simulation
- Efficient index handling of multidimensional periodic boundary conditions
- Atomistic origins of high-performance in hybrid halide perovskite solar cells
- NWChem: Past, Present, and Future
- Quantum embedding theories
- Calculations on millions of atoms with DFT: Linear scaling shows its potential
- ELSI: A Unified Software Interface for Kohn-Sham Electronic Structure Solvers
- Electrostatic considerations affecting the calculated HOMO-LUMO gap in protein molecules
- DFT-FE 1.0: A massively parallel hybrid CPU-GPU density functional theory code using finite-element discretization
- ELSI -- An Open Infrastructure for Electronic Structure Solvers
- Difficulties in applying pure Kohn-Sham density functional theory electronic structure methods to protein molecules
- A TensorFlow Simulation Framework for Scientific Computing of Fluid Flows on Tensor Processing Units
- GPGPU Acceleration of All-Electron Electronic Structure Theory Using Localized Numeric Atom-Centered Basis Functions
- Optimizations of the Eigensolvers in the ELPA Library
- Simulation of quantum many-body dynamics with Tensor Processing Units: Floquet prethermalization
- Generalized canonical purification for density matrix minimization
- Large Scale Distributed Linear Algebra With Tensor Processing Units
- Trends in atomistic simulation software usage
Cited by in corpus (6)
- Roadmap on Electronic Structure Codes in the Exascale Era
- Density Matrix Renormalization Group with Tensor Processing Units
- Solving the Discretised Neutron Diffusion Equations using Neural Networks
- Graph-based Quantum Response Theory and Shadow Born-Oppenheimer Molecular Dynamics
- Self-Supervised Generative Models for Crystal Structures
- Subspace recursive Fermi-operator expansion strategies for large-scale DFT eigenvalue problems on HPC architectures