SearcharxivSearch

arXiv subjects

Anders Johansson

Publications and source records attributed to Anders Johansson.

At least 19 recordsLinked to original sources

Fast and Accurate Foundation Models for Equivariant Machine-Learned Interatomic Potentials

Machine-learned interatomic potentials (MLIPs) have emerged as a transformative tool for computational materials science and chemistry, with universal potentials trained on large and diverse datasets now routinely deployed as 'foundation models' for downstream fine-tuning in targeted chemical spaces. Many scientific applications of the resulting models, such as molecular dynamics (MD), require high inference and training speeds as well as accuracy. In this work, we examine the limits of equivariant MLIPs, which directly encode physical symmetries in model architectures, to achieve these competing targets -- particularly in the regime of extremely large datasets where data efficiency is less critical. We show how this trade-off can be addressed, and present a family of foundation potentials in the NequIP and Allegro equivariant MLIP architectures which achieve leading inference speeds and strong scalability as well as excellent accuracies across a range of community benchmarks -- spanning materials discovery, thermal conductivity prediction, and near-equilibrium mechanical and thermodynamic properties. Accelerations implemented within the NequIP infrastructure now permit training of high-accuracy foundation potentials on ultra-large datasets with dramatically reduced computational cost. Alongside, we show that efforts to improve model accuracy for materials discovery should focus on dataset diversity and improved, consistent descriptions of transition metal compound energy surfaces.

physics.comp-ph

LAMMPS-KOKKOS: Performance Portable Molecular Dynamics Across Exascale Architectures

Since its inception in 1995, LAMMPS has grown to be a world-class molecular dynamics code, with thousands of users, over one million lines of code, and multi-scale simulation capabilities. We discuss how LAMMPS has adapted to the modern heterogeneous computing landscape by integrating the Kokkos performance portability library into the existing C++ code. We investigate performance portability of simple pairwise, many-body reactive, and machine-learned force-field interatomic potentials. We present results on GPUs across different vendors and generations, and analyze performance trends, probing FLOPS throughput, memory bandwidths, cache capabilities, and thread-atomic operation performance. Finally, we demonstrate strong scaling on three exascale machines -- OLCF Frontier, ALCF Aurora, and NNSA El Capitan -- as well as on the CSCS Alps supercomputer, for the three potentials.

cs.DC

Multiscale light-matter dynamics in quantum materials: from electrons to topological superlattices

Light-matter dynamics in topological quantum materials enables ultralow-power, ultrafast devices. A challenge is simulating multiple field and particle equations for light, electrons, and atoms over vast spatiotemporal scales on Exaflop/s computers with increased heterogeneity and low-precision focus. We present a paradigm shift that solves the multiscale/multiphysics/heterogeneity challenge harnessing hardware heterogeneity and low-precision arithmetic. Divide-conquer-recombine algorithms divide the problem into not only spatial but also physical subproblems of small dynamic ranges and minimal mutual information, which are mapped onto best-characteristics-matching hardware units, while metamodel-space algebra minimizes communication and precision requirements. Using 60,000 GPUs of Aurora, DC-MESH (divide-and-conquer Maxwell-Ehrenfest-surface hopping) and XS-NNQMD (excited-state neural-network quantum molecular dynamics) modules of MLMD (multiscale light-matter dynamics) software were 152- and 3,780-times faster than the state-of-the-art for 15.4 million-electron and 1.23 trillion-atom PbTiO3 material, achieving 1.87 EFLOP/s for the former. This enabled the first study of light-induced switching of topological superlattices for future ferroelectric 'topotronics'.

physics.comp-ph

Continuous eigenfunctions of the transfer operator for Dyson models

In this article we address a well known problem at the intersection of ergodic theory and statistical mechanics. We prove that there exists a continuous eigenfunction for the transfer operator corresponding to pair potentials that satisfy a square summability condition on the variations, when the inverse temperature is subcritical. As a corollary we obtain a continuous eigenfunction for the classical Dyson model, with interactions $\J(k)=β\, k^{-α}$, $k\ge1$, in the whole subcritical regime $β<β_c$ for which the parameter $α$ is greater than $3/2$.

math.DS

High-performance training and inference for deep equivariant interatomic potentials

Machine learning interatomic potentials, particularly those based on deep equivariant neural networks, have demonstrated state-of-the-art accuracy and computational efficiency in atomistic modeling tasks like molecular dynamics and high-throughput screening. The size of datasets and demands of downstream workflows are growing rapidly, making robust and scalable software essential. This work presents a major overhaul of the NequIP framework focusing on multi-node parallelism, computational performance, and extensibility. The redesigned framework supports distributed training on large datasets and removes barriers preventing full utilization of the PyTorch 2.0 compiler at train time. We demonstrate this acceleration in a case study by training Allegro models on the SPICE 2 dataset of organic molecular systems. For inference, we introduce the first end-to-end infrastructure that uses the PyTorch Ahead-of-Time Inductor compiler for machine learning interatomic potentials. Additionally, we implement a custom kernel for the Allegro model's most expensive operation, the tensor product. Together, these advancements speed up molecular dynamics calculations on system sizes of practical relevance by up to a factor of 18.

physics.comp-ph

A practical guide to machine learning interatomic potentials -- Status and future

The rapid development and large body of literature on machine learning interatomic potentials (MLIPs) can make it difficult to know how to proceed for researchers who are not experts but wish to use these tools. The spirit of this review is to help such researchers by serving as a practical, accessible guide to the state-of-the-art in MLIPs. This review paper covers a broad range of topics related to MLIPs, including (i) central aspects of how and why MLIPs are enablers of many exciting advancements in molecular modeling, (ii) the main underpinnings of different types of MLIPs, including their basic structure and formalism, (iii) the potentially transformative impact of universal MLIPs for both organic and inorganic systems, including an overview of the most recent advances, capabilities, downsides, and potential applications of this nascent class of MLIPs, (iv) a practical guide for estimating and understanding the execution speed of MLIPs, including guidance for users based on hardware availability, type of MLIP used, and prospective simulation size and time, (v) a manual for what MLIP a user should choose for a given application by considering hardware resources, speed requirements, energy and force accuracy requirements, as well as guidance for choosing pre-trained potentials or fitting a new potential from scratch, (vi) discussion around MLIP infrastructure, including sources of training data, pre-trained potentials, and hardware resources for training, (vii) summary of some key limitations of present MLIPs and current approaches to mitigate such limitations, including methods of including long-range interactions, handling magnetic systems, and treatment of excited states, and finally (viii) we finish with some more speculative thoughts on what the future holds for the development and application of MLIPs over the next 3-10+ years.

cond-mat.mtrl-sci

Atomistic simulations of out-of-equilibrium quantum nuclear dynamics

The rapid advancements in ultrafast laser technology have paved the way for pumping and probing the out-of-equilibrium dynamics of nuclei in crystals. However, interpreting these experiments is extremely challenging due to the complex nonlinear responses in systems where lattice excitations interact, particularly in crystals composed of light atoms or at low temperatures where the quantum nature of ions becomes significant. In this work, we address the nonequilibrium quantum ionic dynamics from first principles. Our approach is general and can be applied to simulate any crystal, in combination with a first-principles treatment of electrons or external machine-learning potentials. It is implemented by leveraging the nonequilibrium time-dependent self-consistent harmonic approximation (TD-SCHA), with a stable, energy-conserving, correlated stochastic integration scheme that achieves an accuracy of $\mathcal{O}(dt^3)$. We benchmark the method with both a simple one-dimensional model to test its accuracy and a realistic 40-atom cell of SrTiO3 under THz laser pump, paving the way for simulations of ultrafast THz-Xray pump-probe spectroscopy like those performed in synchrotron facilities.

cond-mat.other

Ultrafast quantum dynamics in $\mathbf{\mathrm{SrTiO_3}}$ under impulsive THz radiation

Ultrafast spectroscopy paved the way for probing transient states of matter produced through photoexcitation. Despite significant advances, the microscopic processes governing the formation of these states remain largely unknown. This study discloses the nuclear quantum dynamics of $\mathrm{SrTiO_3}$ when excited by THz laser pumping. We use a first-principles machine-learning approach accounting for all atomistic degrees of freedom to examine the time-resolved energy flow across phonon modes following the photoexcitation, revealing the mechanism underpinning the observed phonon upconversion and quantifying the lifetime of the out-of-equilibrium motion. Crucially, our simulations predict that THz pump pulses can generate persistent out-of-equilibrium stress capable of inducing polar order. We observe a correlation between the experimentally measured lifetime of the transient inversion-symmetry-broken state and the duration of the out-of-equilibrium nuclear state. This work not only explains the experimental results on $\mathrm{SrTiO_3}$ but also establishes a framework for simulating the photoexcited quantum dynamics of nuclei from first principles without any empirical input. It lays the groundwork for systematic explorations of complex materials sensitive to photoexcitation.

cond-mat.mtrl-sci

Atomistic evolution of active sites in multi-component heterogeneous catalysts

Multi-component metal nanoparticles (NPs) are of paramount importance in the chemical industry, as most processes therein employ heterogeneous catalysts. While these multi-component systems have been shown to result in higher product yields, improved selectivities, and greater stability through catalytic cycling, the structural dynamics of these materials in response to various stimuli (e.g. temperature, adsorbates, etc.) are not understood with atomistic resolution. Here, we present a highly accurate equivariant machine-learned force field (MLFF), constructed from ab initio training data collected using Bayesian active learning, that is able to reliably simulate PdAu surfaces and NPs in response to thermal treatment as well as exposure to reactive H$_2$ atmospheres. We thus provide a single model that is able to reliably describe the full space of geometric and chemical complexity for such a heterogeneous catalytic system across single crystals, gas-phase interactions, and NPs reacting with H$_2$, including catalyst degradation and explicit reactivity. Ultimately, we provide direct atomistic evidence that verifies existing experimental hypotheses for bimetallic catalyst deactivation under reaction conditions, namely that Pd preferentially segregates into the Au bulk through aggressive catalytic cycling and that this degradation is site-selective, as well as the reactivity for hydrogen exchange as a function of Pd ensemble size. We demonstrate that understanding of the atomistic evolution of these active sites is of the utmost importance, as it allows for design and control of material structure and corresponding performance, which can be vetted in silico.

cond-mat.mtrl-sci

Transferability and Accuracy of Ionic Liquid Simulations with Equivariant Machine Learning Interatomic Potentials

Ionic liquids (ILs) are an exciting class of electrolytes finding applications in many areas from energy storage to solvents, where they have been touted as ``designer solvents'' as they can be mixed to precisely tailor the physiochemical properties. As using machine learning interatomic potentials (MLIPs) to simulate ILs is still relatively unexplored, several questions need to be answered to see if MLIPs can be transformative for ILs. Since ILs are often not pure, but are either mixed together or contain additives, we first demonstrate that a MLIP can be trained to be compositionally transferable, i.e., the MLIP can be applied to mixtures of ions not directly trained on, whilst only being trained on a few mixtures of the same ions. We also investigate the accuracy of MLIPs for a novel IL, which we experimentally synthesize and characterize. Our MLIP trained on $\sim$200 DFT frames is in reasonable agreement with our experiments and DFT.

physics.chem-ph

Unified Differentiable Learning of Electric Response

Predicting response of materials to external stimuli is a primary objective of computational materials science. However, current methods are limited to small-scale simulations due to the unfavorable scaling of computational costs. Here, we implement an equivariant machine-learning framework where response properties stem from exact differential relationships between a generalized potential function and applied external fields. Focusing on responses to electric fields, the method predicts electric enthalpy, forces, polarization, Born charges, and polarizability within a unified model enforcing the full set of exact physical constraints, symmetries and conservation laws. Through application to $α$-SiO$_2$, we demonstrate that our approach can be used for predicting vibrational and dielectric properties of materials, and for conducting large-scale dynamics under arbitrary electric fields at unprecedented accuracy and scale. We apply our method to ferroelectric BaTiO$_3$ and capture the temperature-dependence and time evolution of hysteresis, revealing the underlying microscopic mechanisms of nucleation and growth that govern ferroelectric domain switching.

cond-mat.mtrl-sci

Phase discovery with active learning: Application to structural phase transitions in equiatomic NiTi

Nickel titanium (NiTi) is a protypical shape-memory alloy used in a range of biomedical and engineering devices, but direct molecular dynamics simulations of the martensitic B19' -> B2 phase transition driving its shape-memory behavior are rare and have relied on classical force fields with limited accuracy. Here, we train four machine-learned force fields for equiatomic NiTi based on the LDA, PBE, PBEsol, and SCAN DFT functionals. The models are trained on the fly during NPT molecular dynamics, with DFT calculations and model updates performed automatically whenever the uncertainty of a local energy prediction exceeds a chosen threshold. The models achieve accuracies of 1-2 meV/atom during training and are shown to closely track DFT predictions of B2 and B19' elastic constants and phonon frequencies. Surprisingly, in large-scale molecular dynamics simulations, only the SCAN model predicts a reversible B19' -> B2 phase transition, with the LDA, PBE, and PBEsol models predicting a reversible transition to a previously uncharacterized low-volume phase, which we hypothesize to be a new stable high-pressure phase. We examine the structure of the new phase and estimate its stability on the temperature-pressure phase diagram. This work establishes an automated active learning protocol for studying displacive transformations, reveals important differences between DFT functionals that can only be detected in large-scale simulations, provides an accurate force field for NiTi, and identifies a new phase.

cond-mat.mtrl-sci

Unbiased Atomistic Predictions of Crystal Dislocation Dynamics using Bayesian Force Fields

Crystal dislocation dynamics, especially at high temperatures, represents a subject where experimental phenomenological input is commonly required, and parameter-free predictions, starting from quantum methods, have been beyond reach. This is especially true for phenomena like stacking faults and dislocation cross-slip, which are computationally intractable with methods like density functional theory, as $\sim 10^5-10^6$ atoms are required to reliably simulate such systems. Hence, this work extends quantum-mechanical accuracy to mesoscopic molecular dynamics simulations and opens unprecedented possibilities in material design for extreme mechanical conditions with direct atomistic insight at the deformation mesoscale. To accomplish this, we construct a Bayesian machine-learned force field (MLFF) from ab initio quantum training data, enabling direct observations of high-temperature and high-stress dislocation dynamics in single-crystalline Cu with atomistic resolution. In doing so, a generalizable training protocol is developed for construction of MLFFs for dislocation kinetics, with wide-ranging applicability to other single element systems and alloys. The resulting FLARE MLFF provides excellent predictions of static bulk elastic properties, stacking fault widths and energies, dynamic evolutions and mobilities of edge and screw dislocations, as well as cross-slip energy barriers for screw dislocations, all of which are compared to available experimental measurements. This work marks the first reliable quantitative determination of dislocation mobilities and cross-slip barriers, demonstrating a substantial advancement over previous empirical and machine learned force field models.

cond-mat.mtrl-sci

Learning Interatomic Potentials at Multiple Scales

The need to use a short time step is a key limit on the speed of molecular dynamics (MD) simulations. Simulations governed by classical potentials are often accelerated by using a multiple-time-step (MTS) integrator that evaluates certain potential energy terms that vary more slowly than others less frequently. This approach is enabled by the simple but limiting analytic forms of classical potentials. Machine learning interatomic potentials (MLIPs), in particular recent equivariant neural networks, are much more broadly applicable than classical potentials and can faithfully reproduce the expensive but accurate reference electronic structure calculations used to train them. They still, however, require the use of a single short time step, as they lack the inherent term-by-term scale separation of classical potentials. This work introduces a method to learn a scale separation in complex interatomic interactions by co-training two MLIPs. Initially, a small and efficient model is trained to reproduce short-time-scale interactions. Subsequently, a large and expressive model is trained jointly to capture the remaining interactions not captured by the small model. When running MD, the MTS integrator then evaluates the smaller model for every time step and the larger model less frequently, accelerating simulation. Compared to a conventionally trained MLIP, our approach can achieve a significant speedup (~3x in our experiments) without a loss of accuracy on the potential energy or simulation-derived quantities.

physics.chem-ph

Stability, mechanisms and kinetics of emergence of Au surface reconstructions using Bayesian force fields

Metal surfaces have long been known to reconstruct, significantly influencing their structural and catalytic properties. Many key mechanistic aspects of these subtle transformations remain poorly understood due to limitations of previous simulation approaches. Using active learning of Bayesian machine-learned force fields trained from ab initio calculations, we enable large-scale molecular dynamics simulations to describe the thermodynamics and time evolution of the low-index mesoscopic surface reconstructions of Au (e.g., the Au(111)-`Herringbone,' Au(110)-(1$\times$2)-`Missing-Row,' and Au(100)-`Quasi-Hexagonal' reconstructions). This capability yields direct atomistic understanding of the dynamic emergence of these surface states from their initial facets, providing previously inaccessible information such as nucleation kinetics and a complete mechanistic interpretation of reconstruction under the effects of strain and local deviations from the original stoichiometry. We successfully reproduce previous experimental observations of reconstructions on pristine surfaces and provide quantitative predictions of the emergence of spinodal decomposition and localized reconstruction in response to strain at non-ideal stoichiometries. A unified mechanistic explanation is presented of the kinetic and thermodynamic factors driving surface reconstruction. Furthermore, we study surface reconstructions on Au nanoparticles, where characteristic (111) and (100) reconstructions spontaneously appear on a variety of high-symmetry particle morphologies.

cond-mat.mtrl-sci

Scaling the leading accuracy of deep equivariant models to biomolecular simulations of realistic size

This work brings the leading accuracy, sample efficiency, and robustness of deep equivariant neural networks to the extreme computational scale. This is achieved through a combination of innovative model architecture, massive parallelization, and models and implementations optimized for efficient GPU utilization. The resulting Allegro architecture bridges the accuracy-speed tradeoff of atomistic simulations and enables description of dynamics in structures of unprecedented complexity at quantum fidelity. To illustrate the scalability of Allegro, we perform nanoseconds-long stable simulations of protein dynamics and scale up to a 44-million atom structure of a complete, all-atom, explicitly solvated HIV capsid on the Perlmutter supercomputer. We demonstrate excellent strong scaling up to 100 million atoms and 70% weak scaling to 5120 A100 GPUs.

physics.comp-ph

Uncertainty-aware molecular dynamics from Bayesian active learning for Phase Transformations and Thermal Transport in SiC

Machine learning interatomic force fields are promising for combining high computational efficiency and accuracy in modeling quantum interactions and simulating atomistic dynamics. Active learning methods have been recently developed to train force fields efficiently and automatically. Among them, Bayesian active learning utilizes principled uncertainty quantification to make data acquisition decisions. In this work, we present a general Bayesian active learning workflow, where the force field is constructed from a sparse Gaussian process regression model based on atomic cluster expansion descriptors. To circumvent the high computational cost of the sparse Gaussian process uncertainty calculation, we formulate a high-performance approximate mapping of the uncertainty and demonstrate a speedup of several orders of magnitude. We demonstrate the autonomous active learning workflow by training a Bayesian force field model for silicon carbide (SiC) polymorphs in only a few days of computer time and show that pressure-induced phase transformations are accurately captured. The resulting model exhibits close agreement with both \textit{ab initio} calculations and experimental measurements, and outperforms existing empirical models on vibrational and thermal properties. The active learning workflow readily generalizes to a wide range of material systems and accelerates their computational understanding.

physics.comp-ph