PySEQM 2.0: Accelerated Semiempirical Excited State Calculations on Graphical Processing Units

Abstract

We report the implementation of electronic excited states for semi-empirical quantum chemical methods at the configuration interaction singles (CIS) and time-dependent Hartree-Fock (TDHF) level of theory in the PySEQM software. Built on PyTorch, this implementation leverages GPU acceleration to significantly speed up molecular property calculations. Benchmark tests demonstrate that our approach can compute excited states for molecules with nearly a thousand atoms in under a minute. Additionally, the implementation also includes a machine learning interface to enable parameters re-optimization and neural network training for future machine learning applications for excited state dynamics.