Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
knc6 authored Jun 29, 2024
1 parent 90303df commit 92d9441
Showing 1 changed file with 9 additions and 0 deletions.
9 changes: 9 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
@@ -1,5 +1,7 @@
# 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 Letter, 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

python atomgpt/forward_models/forward_models.py --config_name atomgpt/examples/forward_model/config.json
Expand All @@ -10,4 +12,11 @@ python atomgpt/inverse_models/inverse_models.py --config_name atomgpt/examples/i

#python atomgpt/examples/inverse_model/run.py

# Google colab/Jupyter notebook

[![Open in Google Colab]](https://github.com/knc6/jarvis-tools-notebooks/blob/master/jarvis-tools-notebooks/atomgpt_example.ipynb)

[Open in Google Colab]: https://colab.research.google.com/assets/colab-badge.svg


(Documentation development is in progress...)

0 comments on commit 92d9441

Please sign in to comment.