Zatom-1: Towards a Multimodal Foundation Model for 3D Molecules and Materials
Description
Included are pretrained/finetuned model weights for Zatom-1, a general-purpose architecture for 3D molecules and materials. Also included are some of its generated samples and benchmarking results.
Paper Abstract:
General-purpose 3D modeling in chemistry encompasses molecules and materials, requiring both generative and predictive capabilities. However, most existing AI approaches are optimized for a single domain (molecules or materials) and a single task (generation or prediction), which limits representation sharing and transfer. We introduce Zatom-1, a cross-domain, general-purpose model architecture that unifies generative and predictive learning of 3D molecules and materials. Zatom-1 is a deliberately simplified Transformer trained with a multimodal flow matching objective that jointly models discrete atom types and continuous 3D geometries. This approach supports scalable pretraining with predictable gains as model capacity increases, while enabling fast and stable sampling. We use cross-domain generative pretraining as a universal initialization for downstream multi-task prediction of properties, energies, and forces. Empirically, Zatom-1 outperforms or competes with specialized baselines on both multi-task generative and predictive benchmarks in data-controlled settings, while improving generative inference speed by more than an order of magnitude. Our experiments demonstrate positive predictive transfer between data domains from joint generative pretraining: modeling materials during generative pretraining improves molecular property prediction accuracy. Open-source code and model weights are freely available at https://github.com/Zatom-AI/zatom.
References:
[1] 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*. Zatom-1: Towards a Multimodal Foundation Model for 3D Molecules and Materials. ICLR FM4Science 2026. * denotes equal contribution. Available from: http://arxiv.org/abs/2602.22251
Files
lemat_genbench_jointly_trained_platom_1.zip
Files
(36.7 GB)
| Name | Size | Download all |
|---|---|---|
|
md5:64b760d28f590b1e3fbfaf3fdb4ddcc9
|
11.1 MB | Preview Download |
|
md5:998050e9c7c87ada63fed5a633575f21
|
11.4 MB | Preview Download |
|
md5:8eb122c76316b6697f6eaea9abddbc91
|
11.4 MB | Preview Download |
|
md5:a4f1dfed89040d3bd5c4d86913216416
|
3.9 MB | Preview Download |
|
md5:22a68fe1794569cac3663237e298996c
|
4.1 MB | Preview Download |
|
md5:04b26e1656ec793e8ecc6e747feaf433
|
11.5 MB | Preview Download |
|
md5:39711a546d79d44123ec5f3853917c69
|
11.3 MB | Preview Download |
|
md5:51858170267bdb7b29f294e194b7b467
|
442.1 MB | Download |
|
md5:b6dabd03826a4a312a51abe452a26414
|
442.1 MB | Download |
|
md5:2e6bca003814e438f2e4ff3b2a32eccf
|
676.2 MB | Download |
|
md5:76100f9c03c5f352b065a436b791033d
|
1.2 GB | Download |
|
md5:60f73dd9b9bd619ec97811bfecf70142
|
676.2 MB | Download |
|
md5:347c43df81a7b84c71bf7e5cad6f1cad
|
676.2 MB | Download |
|
md5:33f2ec30bd9d09888adb99c7af3b71a4
|
3.0 GB | Download |
|
md5:439447a01282b3aa0d813282ff79d134
|
676.2 MB | Download |
|
md5:877cbcd3448c080f9158f1c982a6a6a9
|
676.2 MB | Download |
|
md5:785e83130087bc529e5580afcfa9f504
|
676.3 MB | Download |
|
md5:77ac2d7e1433ed973e749ab1466e1cf1
|
3.0 GB | Download |
|
md5:7c69df73169a9fea46be19c2b0f5dfaa
|
1.2 GB | Download |
|
md5:dcedfb7d50cfa827a39ea09ad258b8a3
|
1.2 GB | Download |
|
md5:4082fb0f85f6915ab668ad78c16fc1cb
|
2.6 GB | Download |
|
md5:3387a8a379b68503cceb9fe5141847d1
|
15.6 MB | Preview Download |
|
md5:c51f3daaaf49cf403a706c2b58c307ad
|
13.3 MB | Preview Download |
|
md5:e0d9d25666beebd34b154fc6398dd443
|
7.2 MB | Preview Download |
|
md5:d13310a087bb271ea0293fe35b14a610
|
1.2 GB | Download |
|
md5:ad8b1feb26591599b543c518bf99ce1e
|
676.2 MB | Download |
|
md5:128a240347b76a87202c42233fb205a2
|
1.5 GB | Download |
|
md5:7bf2afc9aa0960d0f4e1ee72b833197e
|
1.2 GB | Download |
|
md5:58aa4a00e7069c2f519005d12bc228fa
|
3.0 GB | Download |
|
md5:ff7d9cfee91c879ab723ac7f02c2385e
|
676.2 MB | Download |
|
md5:074fe5cc88ba094675bb086b40b840bc
|
676.3 MB | Download |
|
md5:3d2a4447b6ab7745db96135cfacc4290
|
1.2 GB | Download |
|
md5:e1bceac71be26c5109d9280c47729bdd
|
1.2 GB | Download |
|
md5:b18f0a86ca56ddd3198670349a31467e
|
1.2 GB | Download |
|
md5:df55cfb75341acb319f3e736c45c2eaf
|
1.8 GB | Download |
|
md5:9336e05d69590211ccd05ae243de52ad
|
4.7 GB | Download |
Additional details
Dates
- Available
-
2026-05-06
Software
- Repository URL
- https://github.com/Zatom-AI/zatom
- Programming language
- Python
- Development Status
- Active