MOFA: Discovering Materials for Carbon Capture with a GenAI- and Simulation-Based Workflow

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Yan, Xiaoli, Hudson, Nathaniel, Park, Hyun, Grzenda, Daniel, Pauloski, J. Gregory, Schwarting, Marcus, Pan, Haochen, Harb, Hassan, Foreman, Samuel, Knight, Chris, Gibbs, Tom, Chard, Kyle, Chaudhuri, Santanu, Tajkhorshid, Emad, Foster, Ian, Moosavi, Mohamad, Ward, Logan, Huerta, E. A.
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913655570825216
author Yan, Xiaoli
Hudson, Nathaniel
Park, Hyun
Grzenda, Daniel
Pauloski, J. Gregory
Schwarting, Marcus
Pan, Haochen
Harb, Hassan
Foreman, Samuel
Knight, Chris
Gibbs, Tom
Chard, Kyle
Chaudhuri, Santanu
Tajkhorshid, Emad
Foster, Ian
Moosavi, Mohamad
Ward, Logan
Huerta, E. A.
author_facet Yan, Xiaoli
Hudson, Nathaniel
Park, Hyun
Grzenda, Daniel
Pauloski, J. Gregory
Schwarting, Marcus
Pan, Haochen
Harb, Hassan
Foreman, Samuel
Knight, Chris
Gibbs, Tom
Chard, Kyle
Chaudhuri, Santanu
Tajkhorshid, Emad
Foster, Ian
Moosavi, Mohamad
Ward, Logan
Huerta, E. A.
contents We present MOFA, an open-source generative AI (GenAI) plus simulation workflow for high-throughput generation of metal-organic frameworks (MOFs) on large-scale high-performance computing (HPC) systems. MOFA addresses key challenges in integrating GPU-accelerated computing for GPU-intensive GenAI tasks, including distributed training and inference, alongside CPU- and GPU-optimized tasks for screening and filtering AI-generated MOFs using molecular dynamics, density functional theory, and Monte Carlo simulations. These heterogeneous tasks are unified within an online learning framework that optimizes the utilization of available CPU and GPU resources across HPC systems. Performance metrics from a 450-node (14,400 AMD Zen 3 CPUs + 1800 NVIDIA A100 GPUs) supercomputer run demonstrate that MOFA achieves high-throughput generation of novel MOF structures, with CO$_2$ adsorption capacities ranking among the top 10 in the hypothetical MOF (hMOF) dataset. Furthermore, the production of high-quality MOFs exhibits a linear relationship with the number of nodes utilized. The modular architecture of MOFA will facilitate its integration into other scientific applications that dynamically combine GenAI with large-scale simulations.
format Preprint
id arxiv_https___arxiv_org_abs_2501_10651
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle MOFA: Discovering Materials for Carbon Capture with a GenAI- and Simulation-Based Workflow
Yan, Xiaoli
Hudson, Nathaniel
Park, Hyun
Grzenda, Daniel
Pauloski, J. Gregory
Schwarting, Marcus
Pan, Haochen
Harb, Hassan
Foreman, Samuel
Knight, Chris
Gibbs, Tom
Chard, Kyle
Chaudhuri, Santanu
Tajkhorshid, Emad
Foster, Ian
Moosavi, Mohamad
Ward, Logan
Huerta, E. A.
Distributed, Parallel, and Cluster Computing
Materials Science
Machine Learning
We present MOFA, an open-source generative AI (GenAI) plus simulation workflow for high-throughput generation of metal-organic frameworks (MOFs) on large-scale high-performance computing (HPC) systems. MOFA addresses key challenges in integrating GPU-accelerated computing for GPU-intensive GenAI tasks, including distributed training and inference, alongside CPU- and GPU-optimized tasks for screening and filtering AI-generated MOFs using molecular dynamics, density functional theory, and Monte Carlo simulations. These heterogeneous tasks are unified within an online learning framework that optimizes the utilization of available CPU and GPU resources across HPC systems. Performance metrics from a 450-node (14,400 AMD Zen 3 CPUs + 1800 NVIDIA A100 GPUs) supercomputer run demonstrate that MOFA achieves high-throughput generation of novel MOF structures, with CO$_2$ adsorption capacities ranking among the top 10 in the hypothetical MOF (hMOF) dataset. Furthermore, the production of high-quality MOFs exhibits a linear relationship with the number of nodes utilized. The modular architecture of MOFA will facilitate its integration into other scientific applications that dynamically combine GenAI with large-scale simulations.
title MOFA: Discovering Materials for Carbon Capture with a GenAI- and Simulation-Based Workflow
topic Distributed, Parallel, and Cluster Computing
Materials Science
Machine Learning
url https://arxiv.org/abs/2501.10651