Saved in:
Bibliographic Details
Main Authors: Huang, Hua, Shao, Wenkai, Hammond, Jeff
Format: Preprint
Published: 2026
Subjects:
Online Access:https://arxiv.org/abs/2601.08077
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866918451374718976
author Huang, Hua
Shao, Wenkai
Hammond, Jeff
author_facet Huang, Hua
Shao, Wenkai
Hammond, Jeff
contents The emergence of artificial intelligence (AI) accelerators like NVIDIA Tensor Cores offers new opportunities to speed up tensor-heavy scientific computations. However, applying them to quantum chemistry is challenging due to strict accuracy demands and irregular data patterns. We propose an adaptive precision algorithm to accelerate the density fitting (DF) method with Gaussian basis sets on AI accelerators using 8-bit integer (INT8) arithmetics. Implemented in the GPU-accelerated PySCF package, the algorithm is tested on more than twenty molecular systems with different NVIDIA GPUs. Compared to the standard FP64 code, our algorithm is up to 204\% faster on a RTX 4090 gaming GPU and up to 364\% faster on a RTX 6000 Ada workstation GPU without compromising the converged energy. This work demonstrates a practical approach to use AI hardware for reliable quantum chemistry simulations.
format Preprint
id arxiv_https___arxiv_org_abs_2601_08077
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Accelerating Density Fitting with Adaptive-precision and 8-bit Integer on AI Accelerators
Huang, Hua
Shao, Wenkai
Hammond, Jeff
Chemical Physics
The emergence of artificial intelligence (AI) accelerators like NVIDIA Tensor Cores offers new opportunities to speed up tensor-heavy scientific computations. However, applying them to quantum chemistry is challenging due to strict accuracy demands and irregular data patterns. We propose an adaptive precision algorithm to accelerate the density fitting (DF) method with Gaussian basis sets on AI accelerators using 8-bit integer (INT8) arithmetics. Implemented in the GPU-accelerated PySCF package, the algorithm is tested on more than twenty molecular systems with different NVIDIA GPUs. Compared to the standard FP64 code, our algorithm is up to 204\% faster on a RTX 4090 gaming GPU and up to 364\% faster on a RTX 6000 Ada workstation GPU without compromising the converged energy. This work demonstrates a practical approach to use AI hardware for reliable quantum chemistry simulations.
title Accelerating Density Fitting with Adaptive-precision and 8-bit Integer on AI Accelerators
topic Chemical Physics
url https://arxiv.org/abs/2601.08077