Taylor-mode automatic differentiation for constructing molecular rovibrational Hamiltonian operators
Atomic and Molecular Clusters
2025-07-29 v3 Chemical Physics
Abstract
We present an automated framework for constructing Taylor series expansions of rovibrational kinetic and potential energy operators for arbitrary molecules, internal coordinate systems, and molecular frame embedding conditions. Expressing operators in a sum-of-products form allows for computationally efficient evaluations of matrix elements in product basis sets. Our approach uses automatic differentiation tools from the Python machine learning ecosystem, particularly the JAX library, to efficiently and accurately generate high-order Taylor expansions of rovibrational operators. The implementation is available at https://github.com/robochimps/vibrojet.
Cite
@article{arxiv.2506.20129,
title = {Taylor-mode automatic differentiation for constructing molecular rovibrational Hamiltonian operators},
author = {Andrey Yachmenev and Emil Vogt and Álvaro Fernández Corral and Yahya Saleh},
journal= {arXiv preprint arXiv:2506.20129},
year = {2025}
}