arXiv · 2411.17011
MOLPIPx: an end-to-end differentiable package for permutationally invariant polynomials in Python and Rust
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 https://github.com/ChemAI-Lab/molpipx.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Manuel S. Drehwald, Asma Jamali, Rodrigo A. Vargas-Hernández. 2024-11-26. MOLPIPx: an end-to-end differentiable package for permutationally invariant polynomials in Python and Rust. https://doi.org/10.1063/5.0250837
Cite the original work for its findings. Save a collection to share your selection of sources.