Large Scale Quantum Chemistry with Tensor Processing Units

被引：19

作者：

Pederson, Ryan ^{[2
,3
,9
]}

Kozlowski, John ^{[1
,2
,3
]}

Song, Ruyi ^{[2
,3
,4
]}

Beall, Jackson ^{[3
,5
]}

Ganahl, Martin ^{[3
]}

Hauru, Markus ^{[3
,6
]}

Lewis, Adam G. M. ^{[3
,5
]}

Yao, Yi ^{[7
]}

Mallick, Shrestha Basu ^{[2
,3
]}

Blum, Volker ^{[4
,7
]}

Vidal, Guifre ^{[2
,3
,8
]}

机构：

[1] Univ Calif Irvine, Dept Chem, Irvine, CA 92617 USA

[2] X Moonshot Factory, Mountain View, CA 94043 USA

[3] SandboxAlphabet, Mountain View, CA 94043 USA

[4] Duke Univ, Dept Chem, Durham, NC 27708 USA

[5] SandboxAQ, Palo Alto, CA 94304 USA

[6] Alan Turing Inst, London NW1 2DB, England

[7] Duke Univ, Thomas Lord Dept Mech Engn & Mat Sci, Durham, NC 27708 USA

[8] Google Quantum AI, Mountain View, CA 94043 USA

[9] Univ Calif Irvine, Dept Phys & Astron, Irvine, CA 92617 USA

来源：

JOURNAL OF CHEMICAL THEORY AND COMPUTATION | 2023年 / 19卷 / 01期

基金：

美国国家科学基金会;

关键词：

DENSITY-FUNCTIONAL CALCULATIONS; PERFORMANCE; ORIGINS;

D O I：

10.1021/acs.jctc.2c00876

中图分类号：

O64 [物理化学（理论化学）、化学物理学];

学科分类号：

070304 ; 081704 ;

摘要：

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.

引用

页码：25 / 32

页数：8

共 50 条

[21] Interactive Quantum Chemistry Enabled by Machine Learning, Graphical Processing Units, and Cloud Computing
Raucci, Umberto
Weir, Hayley
Sakshuwong, Sukolsak
Seritan, Stefan
Hicks, Colton B.
Vannucci, Fabio
Rea, Francesco
Martinez, Todd J.
ANNUAL REVIEW OF PHYSICAL CHEMISTRY, 2023, 74 : 313 - 336
[22] Accelerating large-scale protein structure alignments with graphics processing units
Bin Pang
Nan Zhao
Michela Becchi
Dmitry Korkin
Chi-Ren Shyu
BMC Research Notes, 5 (1)
[23] Exploiting graphical processing units to enable quantum chemistry calculation of large solvated molecules with conductor-like polarizable continuum models
Liu, Fang
Sanchez, David M.
Kulik, Heather J.
Martinez, Todd J.
INTERNATIONAL JOURNAL OF QUANTUM CHEMISTRY, 2019, 119 (01)
[24] Adaptive molecular decomposition: Large-scale quantum chemistry for liquids
Jaervi, Tommi T.
Mayrhofer, Leonhard
Polvi, Jussi
Nordlund, Kai
Pastewka, Lars
Moseler, Michael
JOURNAL OF CHEMICAL PHYSICS, 2013, 138 (10):
[25] Green chemistry for large-scale synthesis of semiconductor quantum dots
Liu, Jin-Hua
Fan, Jun-Bing
Gu, Zheng
Cui, Jing
Xu, Xiao-Bo
Liang, Zhi-Wu
Luo, Sheng-Lian
Zhu, Ming-Qiang
LANGMUIR, 2008, 24 (10) : 5241 - 5244
[26] Building units design and scale chemistry
Férey, G
JOURNAL OF SOLID STATE CHEMISTRY, 2000, 152 (01) : 37 - 48
[27] A TENSOR BASED FRAMEWORK FOR LARGE SCALE SPATIO-TEMPORAL RASTER DATA PROCESSING
Bhattacharya, Sukriti
Braun, Christian
Leopold, Ulrich
FOSS4G 2019 - ACADEMIC TRACK, 2019, 42-4 (W14): : 3 - 9
[28] Tensor voting accelerated by graphics processing units (GPU)
Min, Changki
Medioni, Gerard
18TH INTERNATIONAL CONFERENCE ON PATTERN RECOGNITION, VOL 3, PROCEEDINGS, 2006, : 1103 - +
[29] Density Matrix Renormalization Group with Tensor Processing Units
Ganahl, Martin
Beall, Jackson
Hauru, Markus
Lewis, Adam G. M.
Wojno, Tomasz
Yoo, Jae Hyeon
Zou, Yijian
Vidal, Guifre
PRX QUANTUM, 2023, 4 (01):
[30] Diffusion tensor fiber tracking on graphics processing units
Mittmann, Adiel
Comunello, Eros
von Wangenheim, Aldo
COMPUTERIZED MEDICAL IMAGING AND GRAPHICS, 2008, 32 (07) : 521 - 530

← 1 2 3 4 5 →