Molecular generative models can help navigate the enormous space of chemical compounds for applications in medicine, energy storage, and industrial catalysis. We develop an E(3)-equivariant autoregressive model to generate metal-organic chalcogenide assemblies (MOChAs), crystals containing low-dimensional transition metal chalcogenide structures scaffolded by organic ligands. We aim to predict the organic ligand positions based on the inorganic atoms positions, and to model the periodicity of these crystal structures. Euclidean symmetry-equivariant neural networks (E(3)NNs) like this one are particularly well suited to handling the data types used to describe atomic structures because they account for the Euclidean symmetries found in 3D space: 3D rotation, translation, and inversion.

The intersection of machine learning and physics is fascinating to me, and I look forward to more deeply exploring this area in general, and molecular generative models in particular, over the course of this SuperUROP. As part of Prof. Smidts group, I have had the opportunity to make progress on a generative model for small molecules, and I am excited to continue developing this model for more targeted forms of material design.