Materials Graph Library (MatGL), an open-source graph deep learning library for materials science and chemistry
Tsz Wai Koa✉, Bowen Dengb c, Marcel Nassard, Luis Barroso-Luqueb c, Runze Liua, Ji Qia, Atul C. Thakura, Adesh Rohan Mishraa, Elliott Liua, Gerbrand Cederb c, Santiago Miretd, Shyue Ping Onga✉
- aAiiso Yufeng Li Family Department of Chemical and Nano Engineering, University of California San Diego, La Jolla, CA, USA
- bDepartment of Materials Science and Engineering, University of California Berkeley, Berkeley, CA, USA
- cMaterials Sciences Division, Lawrence Berkeley National Laboratory, Berkeley, CA, USA
- dIntel Labs, Santa Clara, CA, USA
- ✉Corresponding author
Papers open inside SciRah from the original publisher or repository; SciRah never hosts them.
Abstract
Abstract Graph deep learning models, which incorporate a natural inductive bias for atomic structures, 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, MatGL is designed to be an extensible “batteries-included” library for developing advanced model architectures 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 provides several pre-trained foundation potentials (FPs) with coverage of the entire periodic table, and property prediction models for out-of-box usage, benchmarking and fine-tuning. Finally, MatGL integrates with PyTorch Lightning to enable efficient model training.