Skip to content
forked from usnistgov/atomgpt

AtomGPT: Atomistic Generative Pretrained Transformer for Forward and Inverse Materials Design

License

Notifications You must be signed in to change notification settings

wangvei/atomgpt

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

84 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

AtomGPT: atomistic generative pre-trained transformer for forward and inverse materials design

Large language models (LLMs) such as generative pretrained transformers (GPTs) have shown potential for various commercial applications, but their applicability for materials design remains underexplored. In this work, AtomGPT is introduced as a model specifically developed for materials design based on transformer architectures, demonstrating capabilities for both atomistic property prediction and structure generation. This study shows that a combination of chemical and structural text descriptions can efficiently predict material properties with accuracy comparable to graph neural network models, including formation energies, electronic bandgaps from two different methods, and superconducting transition temperatures. Furthermore, AtomGPT can generate atomic structures for tasks such as designing new superconductors, with the predictions validated through density functional theory calculations. This work paves the way for leveraging LLMs in forward and inverse materials design, offering an efficient approach to the discovery and optimization of materials.

Forward model example (structure to property)

python atomgpt/forward_models/forward_models.py --config_name atomgpt/examples/forward_model/config.json

Inverse model example (property to structure)

python atomgpt/inverse_models/inverse_models.py --config_name atomgpt/examples/inverse_model/config.json

Google colab/Jupyter notebook

Open in Google Colab

(Documentation development is in progress...)

AtomGPT schematic

About

AtomGPT: Atomistic Generative Pretrained Transformer for Forward and Inverse Materials Design

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%