neumannrf commited on
Commit
7414592
·
verified ·
1 Parent(s): 44b2d15

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -20,6 +20,6 @@ This repository contains PyTorch source code for loading and performing inferenc
20
  ## Introduction
21
  We present `pos-egnn`, a Position-based Equivariant Graph Neural Network foundation model for Chemistry and Materials. The model was pre-trained on 1.4M samples (i.e., 90%) from the Materials Project Trajectory (MPtrj) dataset to predict energies, forces and stress. `pos-egnn` can be used as a machine-learning potential, as a feature extractor, or can be fine-tuned for specific downstream tasks.
22
 
23
- Besides the model weigths `pos-egnn.v1-6M.pt`, which can be downloaded from [HuggingFace](https://huggingface.co/ibm-research/materials.pos-egnn), we also provide examples for performing inference, feature extraction and molecular dynamics simulation with the model (`example.ipynb`).
24
 
25
  For more information, please reach out to rneumann@br.ibm.com and/or flaviu.cipcigan@ibm.com
 
20
  ## Introduction
21
  We present `pos-egnn`, a Position-based Equivariant Graph Neural Network foundation model for Chemistry and Materials. The model was pre-trained on 1.4M samples (i.e., 90%) from the Materials Project Trajectory (MPtrj) dataset to predict energies, forces and stress. `pos-egnn` can be used as a machine-learning potential, as a feature extractor, or can be fine-tuned for specific downstream tasks.
22
 
23
+ Besides the model weigths `pos-egnn.v1-6M.pt` (download from [HuggingFace](https://huggingface.co/ibm-research/materials.pos-egnn)), we also provide an `example.ipynb` notebook (download from [GitHub](https://github.com/ibm/materials)), which demonstrates how to perform inference, feature extraction and molecular dynamics simulation with the model.
24
 
25
  For more information, please reach out to rneumann@br.ibm.com and/or flaviu.cipcigan@ibm.com