论文标题
大规模量子化学与张量处理单元
Large scale quantum chemistry with Tensor Processing Units
论文作者
论文摘要
我们证明了使用Google的基于云的张量处理单元（TPU）来加速和扩展常规（立方缩放）密度功能理论（DFT）计算。利用512个TPU核心，我们完成了迄今为止最大的此类DFT计算，使用247848个轨道，对应于具有103270电子的10327水分子簇，所有这些都已明确处理。因此，我们的工作铺平了在前所未有的尺度上不受任何系统特异性约束的常规DFT访问和系统使用的道路。
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.
