Search papers, labs, and topics across Lattice.
The paper introduces ai2-kit, a software toolkit designed to streamline AI-accelerated ab initio workflows for complex chemical systems by integrating various computational tasks such as model training, molecular dynamics, and trajectory analysis. This toolkit addresses the limitations of traditional ab initio calculations by providing accessible and reproducible workflows that leverage machine learning potentials, thereby extending the applicability of first-principles data to more complex scenarios. Key results demonstrate the effectiveness of ai2-kit across four applications, showcasing its ability to facilitate active-learning potential construction and enhance property prediction for electrified interfaces and spectroscopic analysis.
AI2 methods can now be transformed from bespoke protocols into reusable workflows, significantly enhancing the efficiency of molecular simulations in complex chemical systems.
Molecular simulations of complex chemical systems, such as catalysis, electrochemistry, and energy storage, often need to capture the interplay of effects such as electronic structure, finite-temperature fluctuations, and electric-field response. Such complexity is difficult to address with traditional ab initio calculations, which are limited by the time and length scales they can reach. AI-accelerated ab initio (AI2) methods use machine learning potentials trained on first-principles data to replace expensive electronic-structure calculations, extending ab initio accuracy to these regimes, but their routine application requires reliable workflows that connect first-principles calculations, model training, molecular dynamics, enhanced sampling, trajectory analysis, and HPC orchestration. Here we present ai2-kit, a software toolkit for developing accessible, reproducible, and extensible AI2 workflows. ai2-kit provides high-semantic-density command-line interfaces and Python APIs for structure and dataset conversion, batch task generation, active-learning screening, job orchestration, and workflow recovery. We demonstrate ai2-kit in four representative applications: active-learning-based machine learning potential construction, free-energy perturbation for redox and acid-base processes, electrochemical machine learning potentials for electrified interfaces, and spectroscopies from machine learning molecular dynamics. ai2-kit also provides AI-agent skills that help users adapt these use cases into customized workflows for their own chemical systems and computational software stacks. Together, ai2-kit helps turn AI2 methods from bespoke computational protocols into reusable and extensible workflows for complex chemical systems, from model construction to property prediction.
