chroma-pytorch

Chroma-PyTorch is an implementation of a generative model for proteins using denoising diffusion probabilistic models and graph neural networks. It aims to facilitate the design of proteins, including those that can bind to specific targets like the spike protein of the coronavirus.