Design-CP: Context Parallelism for Design of Protein Nanoparticles
Design-CP introduces context-parallel inference strategies for RFdiffusion 3 to enable large-scale protein complex design across multi-GPU meshes.
Design-CP addresses memory bottlenecks in generative protein models by implementing 1D row-sharding and 2D grid sharding with ring attention. This allows researchers to model large multimeric complexes without exceeding single-GPU memory limits while maintaining pretrained weights.