Materials Graph Library (MatGL), an open-source graph deep learning library for materials science and chemistry
Preprint 2025 en
Authors
TK
Tsz Wai Ko
BD
Bowen Deng
MN
Marcel Nassar
Abstract
1 min read
Graph deep learning models, which incorporate a natural inductive bias for a collection of atoms, are of immense interest in materials science and chemistry. Here, we introduce the Materials Graph Library (MatGL), an open-source graph deep learning library for materials science and chemistry. Built on top of the popular Deep Graph Library (DGL) and Python Materials Genomics (Pymatgen) packages, our intention is for MatGL to be an extensible ``batteries-included'' library for the development of advanced graph deep learning models for materials property predictions and interatomic potentials. At present, MatGL has efficient implementations for both invariant and equivariant graph deep learning models, including the Materials 3-body Graph Network (M3GNet), MatErials Graph Network (MEGNet), Crystal Hamiltonian Graph Network (CHGNet), TensorNet and SO3Net architectures. MatGL also includes a variety of pre-trained universal interatomic potentials (aka ``foundational materials models (FMM)'') and property prediction models are also included for out-of-box usage, benchmarking and fine-tuning. Finally, MatGL includes support for Pytorch Lightning for rapid training of models.
Tsz Wai Ko, Bowen Deng, Marcel Nassar, Luis Barroso-Luque, Runze Liu, Ji Qi, Atul C. Thakur, Amiya Kanta Mishra, Eric Hsien Lung Liu, Gerbrand Ceder, Santiago Miret, Shyue Ping Ong
Discussion(0)
No comments yet. Be the first to comment.