arrow
返回

Tensor Network State Algorithms on AI Accelerators

delete2024-10-14
delete1
delete
OA
AI
A
Andor Menczer
Ö
Örs Legeza *
DOI:10.1021/acs.jctc.4c00800delete
delete原文链接
delete分享
delete收藏
查看原文
摘要

摘要

En 中文
We introduce novel algorithmic solutions for hybrid CPU-multiGPU tensor network state algorithms utilizing non-Abelian symmetries building on AI-motivated state-of-the-art hardware and software technologies. The presented numerical simulations on the FeMo cofactor, which plays a crucial role in converting atmospheric nitrogen to ammonia, are far beyond the scope of traditional approaches. Our large-scale SU(2) spin adapted density matrix renormalization group calculations up to bond dimension D = 2(16) on complete active space (CAS) size of 18 electrons in 18 orbitals [CAS(18, 18)] demonstrate that the current limit of exact solution, i.e. full-CI limit, can be achieved in fraction of time. Furthermore, benchmarks up to CAS(113, 76) demonstrate the utilization of NVIDIA's highly specialized AI accelerators via NVIDIA Tensor Cores, leading to performance around 115 TFLOPS on a single node supplied with eight NVIDIA A100 devices. As a consequence of reaching 71% of the full capacity of the hardware, the cubic scaling of computational time with bond dimension can be reduced to a linear form for a broad range of D values; thus, breaking the current computational limits of small CAS spaces in ab initio quantum chemistry and material science is becoming a reality. In comparison to strict U(1) implementations with matching accuracy, our solution has an estimated effective performance of 300-500 TFLOPS, which emphasizes the mutual need for both algorithmic and technological developments to push current frontiers on classical computation.
Keyword:
MATRIX RENORMALIZATION-GROUP
QUANTUM-CHEMISTRY
PRODUCT STATES
AI总结

AI总结

对已上传原文的论文进行重点信息的提取,主要内容包括:简要概述、研究摘要、背景介绍、关键亮点、图文解析、展望与总结。

期刊

Journal of Chemical Theory and Computation 封面图
Journal of Chemical Theory and Computation
IF:
5.5
论文数:
1.1W
被引数:
5.4W

机构

H
hun-ren
学者数:
1.2W
论文数: 9.4K
被引数: 14
引用论文

引用论文

err
IF0
err
err0
PREAI
err
err分享
err收藏
NECI: N-Electron Configuration Interaction with an emphasis on state-of-the-art stochastic methods
err2020-07-16
err74
errOAAI
errGuther, Kai; Anderson, Robert J.; Blunt, Nick S.; Bogdanov, Nikolay A.; Cleland, Deidre; Dattani, Nike; Dobrautz, Werner; Ghanem, Khaldoon; Jeszenszki, Peter; Liebermann, Niklas; Li Manni, Giovanni; Lozovoi, Alexander Y.; Luo, Hongjun; Ma, Dongxia; Merz, Florian; Overy, Catherine; Rampp, Markus; Samanta, Pradipta Kumar; Schwarz, Lauretta R.; Shepherd, James J.; Smart, Simon D.; Vitale, Eugenio; Weser, Oskar; Booth, George H.; Alavi, Ali
err分享
err收藏
Density Matrix Renormalization Group with Tensor Processing Units
err2023-02-16
err14
errOAAI
errGanahl, Martin; Beall, Jackson; Hauru, Markus; Lewis, Adam G. M.; Wojno, Tomasz; Yoo, Jae Hyeon; Zou, Yijian; Vidal, Guifre
err分享
err收藏
err
IF0
err
err0
PREAI
err
err分享
err收藏
Matrix product operators, matrix product states, and ab initio density matrix renormalization group algorithms
err2016-07-05
err162
errOAAI
errChan, Garnet Kin-Lic; Keselman, Anna; Nakatani, Naoki; Li, Zhendong; White, Steven R.
err分享
err收藏
学者 查看更多内容