Machine Learning, Artificial Intelligence, Geometric Deep Learning, Drug Discovery, Editors Pick.
employs an equivariant E(n) GNN as a neural network that predicts noise based on input features and time step. At inference time, we first sample the desired number of atoms M, then we can condition EDM on a desired property c, and ask EDM to generate molecules (defined by features x and h) as x, h