Note
The PET-MAD-1.5 models trained for 102 elements at the r2SCAN level of theory are now available! These models are more robust, more accurate and faster than the previous PET-MAD models. We highly recommend using these models for all applications, especially molecular dynamics simulations. Try them out and let us know what you think!
from upet.calculator import UPETCalculator
calculator = UPETCalculator(model="pet-mad-s", version="1.5.0", device="cuda")Note
Are you here to try our Matbench model? Here's all you need. This model is excellent for convex hull energies, geometry optimization and phonons, but we highly recommend the lighter and more universal PET-MAD for molecular dynamics!
from upet.calculator import UPETCalculator
calculator = UPETCalculator(model="pet-oam-xl", version="1.0.0", device="cuda")Warning
This repository is a successor of the PET-MAD repository, which is now deprecated.
The package has been renamed to UPET to reflect the broader scope of the models
and functionalities provided, that go beyond the original PET-MAD model.
Please use version 1.4.4 of PET-MAD package if you want to use the old API.
The migration guide and the
older version of the README are available in the repository.
UPET is a family of universal interatomic potentials for advanced materials modeling across the periodic table. These models are based on the Point Edge Transformer (PET) architecture trained on various popular atomistic datasets, and they are capable of predicting energies and forces in complex atomistic workflows. The package also ships PET-MAD-DOS, a universal model for predicting the electronic density of states (DOS) of materials and molecules, as well as their Fermi levels and bandgaps.
📖 Full documentation: https://lab-cosmo.github.io/upet/latest/
Install UPET from PyPI:
pip install upetOr directly from the GitHub repository:
pip install upet@git+https://github.com/lab-cosmo/upet.gitSee the installation guide
for additional methods (specific versions, uv, etc.).
Run a single-point evaluation with the ASE-compatible UPETCalculator:
from upet.calculator import UPETCalculator
from ase.build import bulk
atoms = bulk("Si", cubic=True, a=5.43, crystalstructure="diamond")
calculator = UPETCalculator(model="pet-mad-s", version="1.5.0", device="cpu")
atoms.calc = calculator
energy = atoms.get_potential_energy()
forces = atoms.get_forces()The first call downloads the checkpoint from the
HuggingFace repository and caches it
locally, so subsequent calls are fast. If version is omitted, the latest
available version is used.
To list available models, sizes, and versions:
from upet import list_upet
list_upet(model="pet-mad", size="s")
list_upet(model="pet-mad")
list_upet()UPET ships several pre-trained model families (PET-MAD, PET-OAM, PET-OMat, PET-OMATPES, PET-SPICE) at multiple sizes. The full table with supported elements, training sets, and recommended use cases is in the models documentation. All checkpoints are available on the HuggingFace repository.
The documentation covers the complete feature surface:
- ASE workflows: non-conservative forces, uncertainty quantification, rotational averaging, dispersion corrections, and PET-MAD-DOS for DOS / Fermi levels / bandgaps.
- Batched evaluation with metatrain.
- LAMMPS (including KOKKOS GPU support), i-PI, TorchSim, and GROMACS interfaces.
- Fine-tuning and example gallery.
- FAQ and known issues.
- FAQ and known issues.
More worked examples for ASE, i-PI, and LAMMPS are also available in the Atomistic Cookbook.
If you found our models useful, please cite the corresponding articles. The full list with copy-pasteable BibTeX is also available in the documentation.
PET-MAD-1.5:
@misc{PET-MAD-1.5-2026,
title={High-quality, high-information datasets for universal atomistic machine learning},
author={Cesare Malosso and Filippo Bigi and Paolo Pegolo and Joseph W. Abbott and Philip Loche and Mariana Rossi and Michele Ceriotti and Arslan Mazitov},
year={2026},
eprint={2603.02089},
archivePrefix={arXiv},
primaryClass={cond-mat.mtrl-sci},
url={https://arxiv.org/abs/2603.02089},
}Current UPET architecture, PET-OAM, PET-OMat, PET-OMAD, PET-OMATPES, PET-SPICE:
@misc{pushing-unconstrained-2026,
title={Pushing the limits of unconstrained machine-learned interatomic potentials},
author={Filippo Bigi and Paolo Pegolo and Arslan Mazitov and Michele Ceriotti},
year={2026},
eprint={2601.16195},
archivePrefix={arXiv},
primaryClass={physics.chem-ph},
url={https://arxiv.org/abs/2601.16195},
}PET-MAD:
@misc{PET-MAD-2025,
title={PET-MAD as a lightweight universal interatomic potential for advanced materials modeling},
author={Mazitov, Arslan and Bigi, Filippo and Kellner, Matthias and Pegolo, Paolo and Tisi, Davide and Fraux, Guillaume and Pozdnyakov, Sergey and Loche, Philip and Ceriotti, Michele},
journal={Nature Communications},
volume={16},
number={1},
pages={10653},
year={2025},
url={https://doi.org/10.1038/s41467-025-65662-7},
}PET-MAD-DOS:
@misc{PET-MAD-DOS-2025,
title={A universal machine learning model for the electronic density of states},
author={Wei Bin How and Pol Febrer and Sanggyu Chong and Arslan Mazitov and Filippo Bigi and Matthias Kellner and Sergey Pozdnyakov and Michele Ceriotti},
year={2025},
eprint={2508.17418},
archivePrefix={arXiv},
primaryClass={physics.chem-ph},
url={https://arxiv.org/abs/2508.17418},
}For a general citation for the PET architecture, you can use
@misc{PET-ECSE-2023,
title = {Smooth, Exact Rotational Symmetrization for Deep Learning on Point Clouds},
journal = {Advances in {{Neural Information Processing Systems}}},
author = {Pozdnyakov, Sergey and Ceriotti, Michele},
year = 2023,
volume = {36},
pages = {79469--79501},
}This project is maintained by @abmazitov and @frostedoyster, who will reply to issues and pull requests opened on this repository as soon as possible. You can mention them directly if you have not received an answer after a couple of days.