Update README.md
Browse files
README.md
CHANGED
|
@@ -14,4 +14,69 @@ pipeline_tag: graph-ml
|
|
| 14 |
tags:
|
| 15 |
- chemistry
|
| 16 |
- biology
|
| 17 |
-
--
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 14 |
tags:
|
| 15 |
- chemistry
|
| 16 |
- biology
|
| 17 |
+
- foundation-model
|
| 18 |
+
- generative-model
|
| 19 |
+
- predictive-model
|
| 20 |
+
- representation-learning
|
| 21 |
+
- transformer
|
| 22 |
+
- molecule
|
| 23 |
+
- material
|
| 24 |
+
- property
|
| 25 |
+
- energy
|
| 26 |
+
- forces
|
| 27 |
+
- mlip
|
| 28 |
+
---
|
| 29 |
+
|
| 30 |
+
<div align="center">
|
| 31 |
+
|
| 32 |
+
# Zatom-1
|
| 33 |
+
|
| 34 |
+
[](https://arxiv.org/abs/2602.22251)
|
| 35 |
+
|
| 36 |
+
<a href="https://arxiv.org/abs/2602.22251"><img src="zatom_1.png" width="600"></a>
|
| 37 |
+
|
| 38 |
+
</div>
|
| 39 |
+
|
| 40 |
+
This repository contains the model weights for Zatom-1, as introduced in [Zatom-1: A Multimodal Flow Foundation Model for 3D Molecules and Materials](https://huggingface.co/papers/2602.22251).
|
| 41 |
+
|
| 42 |
+
## GitHub repository
|
| 43 |
+
|
| 44 |
+
https://github.com/Zatom-AI/zatom
|
| 45 |
+
|
| 46 |
+
## Open-source resources
|
| 47 |
+
|
| 48 |
+
Zatom-1 builds upon the source code and data from the following projects:
|
| 49 |
+
|
| 50 |
+
- [all-atom-diffusion-transformer](https://github.com/facebookresearch/all-atom-diffusion-transformer)
|
| 51 |
+
- [flow_matching](https://github.com/facebookresearch/flow_matching)
|
| 52 |
+
- [flowmm](https://github.com/facebookresearch/flowmm)
|
| 53 |
+
- [jvp_flash_attention](https://github.com/amorehead/jvp_flash_attention)
|
| 54 |
+
- [lemat-genbench](https://github.com/LeMaterial/lemat-genbench)
|
| 55 |
+
- [lightning-hydra-template](https://github.com/ashleve/lightning-hydra-template)
|
| 56 |
+
- [PlatonicTransformers](https://github.com/niazoys/PlatonicTransformers)
|
| 57 |
+
- [ProteinWorkshop](https://github.com/a-r-j/ProteinWorkshop)
|
| 58 |
+
- [posebusters](https://github.com/maabuu/posebusters)
|
| 59 |
+
- [tabasco](https://github.com/carlosinator/tabasco)
|
| 60 |
+
|
| 61 |
+
We thank all their contributors and maintainers!
|
| 62 |
+
|
| 63 |
+
## Acknowledgements
|
| 64 |
+
|
| 65 |
+
This research used resources of the National Energy Research Scientific Computing Center, a DOE Office of Science User Facility supported by the Office of Science of the U.S. Department of Energy under Contract No. DE-AC02-05CH11231 using AI4Sci@NERSC award NERSC DDR-ERCAP0036206 awarded to AM. NBE would like to acknowledge support from the U.S. Department of Energy, Office of Science, Office of Advanced Scientific Computing Research, EXPRESS: 2025 Exploratory Research for Extreme-Scale Science program, and the Scientific Discovery through Advanced Computing (SciDAC) program, under Contract Number DE-AC02-05CH11231 at Berkeley Lab.
|
| 66 |
+
|
| 67 |
+
## Citation
|
| 68 |
+
|
| 69 |
+
If you use the code or data associated with this package or otherwise find this work useful, please cite:
|
| 70 |
+
|
| 71 |
+
```bibtex
|
| 72 |
+
@article{zatom_1_2026,
|
| 73 |
+
title={Zatom-1: A Multimodal Flow Foundation Model for 3D Molecules and Materials},
|
| 74 |
+
author={Alex Morehead* and Miruna Cretu* and Antonia Panescu* and Rishabh Anand* and Maurice Weiler* and Tynan Perez* and Samuel Blau and Steven Farrell and Wahid Bhimji and Anubhav Jain and Hrushikesh Sahasrabuddhe and Pietro Li貌 and Tommi Jaakkola and Rafael G贸mez-Bombarelli and Rex Ying* and Ben Erichson* and Michael Mahoney*},
|
| 75 |
+
year={2026},
|
| 76 |
+
eprint={2602.22251},
|
| 77 |
+
archivePrefix={arXiv},
|
| 78 |
+
primaryClass={cs.LG},
|
| 79 |
+
url={https://arxiv.org/abs/2602.22251},
|
| 80 |
+
note={* denotes equal contribution}
|
| 81 |
+
}
|
| 82 |
+
```
|