arXiv · 2611.02315
GPU infrastructure for ML/MM with electrostatic embedding for enzyme chemistry
Abstract
We present a fully GPU-native framework for machine learning/molecular mechanics (ML/MM) enhanced sampling simulations with electrostatic embedding, demonstrated on an enzymatic reaction. The framework combines a JAX port of the pretrained MACE-POLAR-1-M model, an electrostatic embedding scheme compatible with the MACE-POLAR architecture, and integration with JAX-MD and PLUMED-JAX for end-to-end compiled enhanced sampling. The implementation is developed using GPT-6.1-Sol within Codex, with reasoning effort set to high. Using replica exchange umbrella sampling, we calculate the free energy profile of nucleophilic substitution in haloalkane dehalogenase in approximately 200 H100 GPU hours, treating the entire protein and reactants with MACE-POLAR-1-M-JAX and most of the solvent with molecular mechanics.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Venkat Kapil. 2026-11-02. GPU infrastructure for ML/MM with electrostatic embedding for enzyme chemistry. https://arxiv.org/abs/2611.02315
Cite the original work for its findings. Save a collection to share your selection of sources.