Add img

README.md

@@ -28,6 +28,8 @@ All-atom Diffusion Transformers (ADiTs) jointly generate both periodic materials
 - An autoencoder maps a unified, all-atom representations of molecules and materials to a shared latent embedding space; and
 - A diffusion model is trained to generate new latent embeddings that the autoencoder can decode to sample new molecules or materials.
 
+![](https://github.com/facebookresearch/all-atom-diffusion-transformer/blob/fc4364c86531db395153e37707ec692dfd3408ec/ADiT.png)
+
 ## Citation
 
 ArXiv link: [*All-atom Diffusion Transformers: Unified generative modelling of molecules and materials*](https://www.arxiv.org/abs/2503.03965)