Implementation of Chroma, generative models of protein using DDPM and GNNs, in Pytorch - View it on GitHub