The electronic structure calculations remain a major bottleneck in ab initio nonadiabatic molecular dynamics. We develop an efficient TDDFT-based FSSH implementation in the GPU4PySCF package for medium-sized molecular systems. Our approach combines density fitting, TDDFT with minimal auxiliary basis sets (TDDFT-ris), and an approximate Z-vector solver to reduce the computational cost of TDDFT excited states and derivative coupling calculations. These approximations introduce negligible errors in realistic FSSH workloads while maintaining high computational efficiency. Benchmark results show that, for 73-atom systems with a triple-ζ basis set, individual electronic structure calculations are completed within one minute on a single NVIDIA A100 GPU.
@article{arxiv.2605.06489,
title = {TDDFT Gradients and Nonadiabatic Couplings with Minimal Auxiliary Basis Set Approximation for Fewest-Switches Surface Hopping Dynamics},
author = {Cheng Fan and Zhichen Pu and Zehao Zhou and Yuanheng Wang and Yi Qin Gao and Qiming Sun},
journal= {arXiv preprint arXiv:2605.06489},
year = {2026}
}