Published May 6, 2026 | Version 1.0.2
Model Open

Zatom-1: Towards a Multimodal Foundation Model for 3D Molecules and Materials

  • 1. ROR icon University of Missouri

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