Mixed-precision ab initio tensor network state methods adapted for NVIDIA Blackwell technology via emulated FP64 arithmetic

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Brower, Cole, Bernabeu, Samuel Rodriguez, Hammond, Jeff, Gunnels, John, Xanthea, Sotiris S., Ganahl, Martin, Menczer, Andor, Legeza, Örs
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866909826872770560
author Brower, Cole
Bernabeu, Samuel Rodriguez
Hammond, Jeff
Gunnels, John
Xanthea, Sotiris S.
Ganahl, Martin
Menczer, Andor
Legeza, Örs
author_facet Brower, Cole
Bernabeu, Samuel Rodriguez
Hammond, Jeff
Gunnels, John
Xanthea, Sotiris S.
Ganahl, Martin
Menczer, Andor
Legeza, Örs
contents We report cutting-edge performance results via mixed-precision spin adapted ab initio Density Matrix Renormalization Group (DMRG) electronic structure calculations utilizing the Ozaki scheme for emulating FP64 arithmetic through the use of fixed-point compute resources. By approximating the underlying matrix and tensor algebra with operations on a modest number of fixed-point representatives (``slices''), we demonstrate on smaller benchmark systems and for the active compounds of the FeMoco and cytochrome P450 (CYP) enzymes with complete active space (CAS) sizes of up to 113 electrons in 76 orbitals [CAS(113, 76)] and 63 electrons in 58 orbitals [CAS(63, 58)], respectively, that the chemical accuracy can be reached with mixed-precision arithmetic. We also show that, due to its variational nature, DMRG provides an ideal tool to benchmark accuracy domains, as well as the performance of new hardware developments and related numerical libraries. Detailed numerical error analysis and performance assessment are also presented for subcomponents of the DMRG algebra by systematically interpolating between double- and pseudo-half-precision. Our analyis represents the first quantum chemistry evaluation of FP64 emulation for correlated calculations capable of achieving chemical accuracy and emulation based on fixed-point arithmetic, and it paves the way for the utilization of state-of-the-art Blackwell technology in tree-like tensor network state electronic structure calculations, opening new research directions in materials sciences and beyond.
format Preprint
id arxiv_https___arxiv_org_abs_2510_04795
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Mixed-precision ab initio tensor network state methods adapted for NVIDIA Blackwell technology via emulated FP64 arithmetic
Brower, Cole
Bernabeu, Samuel Rodriguez
Hammond, Jeff
Gunnels, John
Xanthea, Sotiris S.
Ganahl, Martin
Menczer, Andor
Legeza, Örs
Chemical Physics
Strongly Correlated Electrons
Computational Physics
We report cutting-edge performance results via mixed-precision spin adapted ab initio Density Matrix Renormalization Group (DMRG) electronic structure calculations utilizing the Ozaki scheme for emulating FP64 arithmetic through the use of fixed-point compute resources. By approximating the underlying matrix and tensor algebra with operations on a modest number of fixed-point representatives (``slices''), we demonstrate on smaller benchmark systems and for the active compounds of the FeMoco and cytochrome P450 (CYP) enzymes with complete active space (CAS) sizes of up to 113 electrons in 76 orbitals [CAS(113, 76)] and 63 electrons in 58 orbitals [CAS(63, 58)], respectively, that the chemical accuracy can be reached with mixed-precision arithmetic. We also show that, due to its variational nature, DMRG provides an ideal tool to benchmark accuracy domains, as well as the performance of new hardware developments and related numerical libraries. Detailed numerical error analysis and performance assessment are also presented for subcomponents of the DMRG algebra by systematically interpolating between double- and pseudo-half-precision. Our analyis represents the first quantum chemistry evaluation of FP64 emulation for correlated calculations capable of achieving chemical accuracy and emulation based on fixed-point arithmetic, and it paves the way for the utilization of state-of-the-art Blackwell technology in tree-like tensor network state electronic structure calculations, opening new research directions in materials sciences and beyond.
title Mixed-precision ab initio tensor network state methods adapted for NVIDIA Blackwell technology via emulated FP64 arithmetic
topic Chemical Physics
Strongly Correlated Electrons
Computational Physics
url https://arxiv.org/abs/2510.04795