PaddleMaterials

🚀 Introduction

PaddleMaterials is an end-to-end AI4Materials toolkit built on the PaddlePaddle deep learning framework. Designed as a data-mechanism dual-driven platform for developing and deploying foundation models in materials science, PPMat enables researchers to efficiently build AI models and accelerate material discovery using pretrained models.

🧩 Core Capabilities

Task	Description	Typical Applications
Property Prediction (PP)	Predict material properties from structure	Forward design or predict formation energy, band gap, elastic moduli etc.
Structure Generation (SG)	Generate novel crystal structures	Inverse design or structure generation
Machine Learning Interatomic Potential (MLIP)	Surrogate Model for DFT as ML potentials	Molecular dynamics simulations
Electronic Structure (ES)	Surrogate Model for DFT to predict physical field	Predict electronic density
Spectrum Elucidation (SE)	Reconstruct structures from spectra	NMR structure elucidation
Spectrum Enhancement (SPEN)	Enhance microscopy and spectrum signals	STEM image enhancement, denoising

🧱 Supported Materials

• Inorganic Crystals - Well-supported with multiple datasets and pretrained models
• Organic Molecules - Support for small molecule datasets and property prediction
• Polymers, catalysts, and amorphous materials are under development

✨ Why PaddleMaterials?

• ✅ Rich Pretrained Models & AI-ready Datasets - 50+ pretrained models ready for inference and Multiple curated datasets for training
• ✅ Multi-Task Integration - Unified framework across tasks of PP, SG, MLIP, ES, SE, SPEN etc.
• ✅ Multi-Hardware Support - Full support for NVIDIA GPUs and MetaX GPUs and Intel CPUs
• ✅ Production-Ready - Easy to use with standandlize design & distributed training, mixed precision, checkpoint recovery

📑 Support Tasks

Task	Description	Link
Property Prediction (PP)	Predict formation energy, band gap, elastic properties	README
Structure Generation (SG)	Generate new crystal structures with diffusion models	README
Machine Learning Interatomic Potential (MLIP)	DFT-accurate potentials for molecular dynamics	README
Electronic Structure (ES)	Predict electronic structure properties	README
Spectrum Elucidation (SE)	Reconstruct molecular structures from NMR spectra	README
Spectrum Enhancement (SPEN)	Enhance microscopy and spectral signals	README

🤖 Available Pretrained Models

Task	Models	Dataset
Property Prediction	MEGNet, iComformer, DimeNet++	MP2018, MP2024, JARVIS
Structure Generation	MatterGen, DiffCSP	MP20, ALEX
Machine Learning Interatomic Potential	CHGNet, MatterSim	MPTRJ
Electronic Structure	InfGCN	QM9_ES, MP_ES, OMol25_MC_ES
Spectrum Elucidation	DiffNMR	MSD_NMR
Spectrum Enhancement	SFIN	SFIN-HAADF/BF

Full model list: See MODEL_REGISTRY

---

🚀 Get Started

🔧 Installation

Please refer to the installation document for your hardware environment. See SupportedHardwareList for more multi-hardware adaptation information.

---

⚡ Easy Inference

Property Prediction

Predict material formation energy using a pretrained MEGNet model:

python property_prediction/predict.py \
    --model_name='megnet_mp2018_train_60k_e_form' \
    --weights_name='best.pdparams' \
    --cif_file_path='./property_prediction/example_data/cifs/' \
    --save_path='result.csv'

Structure Generation

Generate novel crystal structures:

python structure_generation/predict.py \
    --model_name='mattergen_mp20' \
    --num_structures=100 \
    --save_path='generated_structures/'

Interatomic Potentials

Run molecular dynamics with ML potentials:

python interatomic_potentials/run_md.py 
    --model_name='mattersim_1M' 
    --structure_path='input.cif' 
    --temperature=300

Electronic Structure

Run prediction of elcutorninc density:

python interatomic_potentials/run_md.py 
    --model_name='mattersim_1M' 
    --structure_path='input.cif' 
    --temperature=300

Spectrum Elucidation

Run NMR spectrum elucidate:

python spectrum_elucidation/sample.py 
    --config_path='spectrum_elucidation/configs/diffnmr/DiffNMR.yaml' 
    --weights_name='DiffNMR_nless15_best.pdparams' 
    --save_path='result_diffnmr_nless15/' 
    --checkpoint_path="pretrained"

Spectrum Enhancement

Run prediction of elcutorninc density:

python spectrum_enhancement/predict.py 
    --model_name sfin_haadf_enhance 
    --split val

---

🏋️ Start Training

For training and fine-tuning, refer to the documentation.

---

🤝 Contributors & Cooperation & Community

Thanks to all contributors who have helped build PaddleMaterials！

Thanks for the following organiziton for cooprative support!

Join the PaddleMaterials WeChat group to discuss with us!

🛠️ Contribute to PaddleMaterials

For developer, please refer to architecture.

---

📜 License

PaddleMaterials is licensed under the Apache License 2.0.

---

🎓 Citation

@misc{paddlematerials2025,
  title={PaddleMaterials, a deep learning toolkit based on PaddlePaddle for material science.},
  author={PaddleMaterials Contributors},
  howpublished = {\url{https://github.com/PaddlePaddle/PaddleMaterials}},
  year={2025}
}

---

🙏 Acknowledgements

This repository references code from the following projects:

PaddleScience | Matgl | CDVAE | DiffCSP | MatterGen | MatterSim | CHGNet | AIRS