MOLPIPx: an end-to-end differentiable package for permutationally invariant polynomials in Python and Rust

Manuel S. Drehwald,Asma Jamali,Rodrigo A. Vargas-Hernández
2024-11-26
Abstract:In this work, we present MOLPIPx, a versatile library designed to seamlessly integrate Permutationally Invariant Polynomials (PIPs) with modern machine learning frameworks, enabling the efficient development of linear models, neural networks, and Gaussian process models. These methodologies are widely employed for parameterizing potential energy surfaces across diverse molecular systems. MOLPIPx leverages two powerful automatic differentiation engines -JAX and EnzymeAD-Rust- to facilitate the efficient computation of energy gradients and higher-order derivatives, which are essential for tasks such as force field development and dynamic simulations. MOLPIPx is available at <a class="link-external link-https" href="https://github.com/ChemAI-Lab/molpipx" rel="external noopener nofollow">this https URL</a>.
Chemical Physics,Computational Physics
What problem does this paper attempt to address?