cs.LGJul 3, 2026

Design-CP: Context Parallelism for Design of Protein Nanoparticles

Authors: Lorenzo TarriconeHelen E. EisenachAiko MuraishiCharlotte M. Deane

Abstract

Many all-atom generative protein models can in principle design large multimeric complexes by jointly modelling all chains, but their quadratic token- and atom-pair representations quickly exceed single-GPU memory as the number of chains and residues modelled grows. We introduce Design-CP, two context-parallel (CP) inference strategies for RFdiffusion 3 (1D row-sharding and 2D grid sharding with ring attention) that distribute the quadratic activations across a multi-GPU mesh while preserving pretrained weights. We characterise their scaling when sampling icosahedral assemblies, showing that the maximum feasible asymmetric subunit (ASU) size grows with the expected square-root trend in GPU count and that 2D sharding achieves better wall-clock scaling. Moreover, we show how strong point-group symmetry constraints make CP usable out of the box for end-to-end, all-atom design of icosahedral nanoparticles, yielding favourable in silico structural and interface metrics. Finally, we demonstrate octahedral nanoparticle design on a small cluster of workstation-grade 16GB GPUs, illustrating how Design-CP can be a practical path towards democratising large-assembly protein design.

Explore similar work

CardsList
  1. Emyx: Fast and efficient all-atom protein generation

    Jun 12, 2026Nicholas J. Williams, Ward Haddadin, Matteo P. Ferla +6Protein DesignProtein

  2. SPID: Distilled Protein Backbone Generation

    Oct 3, 2025Liyang Xie, Haoran Zhang, Zhendong Wang +2Protein Design