exa-AMD: An Exascale-Ready Framework for Accelerating the Discovery and Design of Functional Materials

ORAL

Abstract

We present exa-AMD, an open-source, high-performance framework designed for accelerated materials discovery on modern supercomputers. exa-AMD overcomes key computational bottlenecks in large-scale structure prediction through task-based parallelization, adaptive load balancing, and optimized data management for CPU and GPU architectures. The framework automates the end-to-end workflow—from generating candidate structures to evaluating formation energies and updating phase diagrams. Its modular design allows users to easily replace or extend components with custom machine learning models, alternative initial structure templates, and future structure generators, enabling flexible integration with emerging AI approaches. We demonstrate strong scaling across high-performance computing platforms and highlight applications to Na–B–C, Ce–Co–B, and Fe–Co–Zr systems, establishing exa-AMD as a robust and exascale-ready tool for accelerating the discovery and design of functional materials.

*This work was supported by the U.S. Department of Energy (DOE), Office of Science, Basic Energy Sciences, Materials Science and Engineering Division through the Computational Material Science Center program. Ames National Laboratory is operated for the U.S. DOE by Iowa State University under contract # DE-AC02-07CH11358. Los Alamos National Laboratory is operated by Triad National Security, LLC, for the National Nuclear Security Administration of U.S. Department of Energy under Contract No. 89233218CNA000001. This research used resources provided by the National Energy Research Scientific Computing Center, supported by the Office of Science of the U.S. Department of Energy under Contract No. DE-AC02-05CH11231, and resources provided by the Los Alamos National Laboratory Institutional Computing Program. Any opinions, findings, and conclusions or recommendations expressed in this material are those of the authors and do not necessarily reflect the views of the U.S. Department of Energy's National Nuclear Security Administration.

Presenters

  • Weiyi Xia

    • Ames National Laboratory

Authors

  • Weiyi Xia

    • Ames National Laboratory
  • Maxim Moraru

    • Los Alamos National Laboratory
  • Ying Wai Li

    • Los Alamos National Laboratory
  • Cai-Zhuang Wang

    • Iowa State University