chroma-pytorch
Implementation of Chroma, generative models of protein using DDPM and GNNs, in Pytorch
An open-source AI project that designs entirely new proteins from scratch using generative models, recreating a system originally built by Generate Biomedicines for digital protein design.
Chroma-pytorch is an open-source attempt to recreate a powerful AI system called Chroma, originally developed by the biotech company Generate Biomedicines. Chroma's purpose is to design entirely new proteins from scratch. Proteins are the molecular machines that drive biology, and being able to design them on demand has massive implications for creating new medicines, vaccines, and materials. Instead of waiting to discover useful proteins in nature, this software aims to generate them digitally.
The system works using a type of artificial intelligence called a "generative model," which is similar to the technology behind AI image generators. It learns the complex 3D rules of how proteins fold and function by studying vast amounts of biological data. To create a new protein, it starts with random molecular noise and gradually refines it step-by-step into a stable, realistic protein structure. The project uses a combination of neural network techniques to understand the intricate spatial relationships between thousands of atoms.
Researchers, biologists, and machine learning engineers would use this to design custom proteins for specific jobs. For example, as noted in the project's documentation, you could theoretically design a protein that binds directly to the coronavirus spike protein to neutralize it, or create entirely new protein-based drugs. It gives scientists a way to prototype biological solutions digitally before ever stepping into a physical lab.
The project is an early-stage, work-in-progress effort to make advanced protein design accessible to the broader research community. The creator notes that this approach to protein design is an emerging field, with competing academic labs exploring similar methods. It is built in Pytorch, a popular programming tool for AI, and the developer plans to integrate additional scientific language models as the project evolves. The effort is tied to OpenBioML, a broader initiative to keep cutting-edge biological AI freely available to the public.
Where it fits
- Design a protein that binds to a specific target like a virus spike protein.
- Generate novel protein-based drug candidates digitally before lab testing.
- Prototype custom proteins for new vaccines or materials research.