amorehead commited on
Commit
dc319bd
verified
1 Parent(s): 201ee70

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +66 -1
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
+ [![Paper](https://img.shields.io/badge/arXiv-2602.22251-b31b1b.svg)](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
+ ```