How Claude is uplifting biomolecular modeling
Key Points
- 1Researchers utilized Claude to optimize over 30 open-source biomolecular models, achieving a 4x average speed improvement and introducing a low-memory mode for modeling large protein systems on a single GPU.
- 2The project developed custom "FlashPairformer" kernels to accelerate critical attention operations, enabling the prediction of massive molecular structures that were previously restricted by significant computational resource requirements.
- 3These optimized models successfully streamlined de novo protein binder design, reducing the required GPU compute by two orders of magnitude while maintaining high performance, a milestone now being celebrated through a collaborative protein design competition.
This report details how Claude, an AI model, successfully optimized over 30 open-source biomolecular modeling tools, significantly enhancing both computational speed and memory efficiency for protein structure prediction and design.
Core Methodology and Technical Optimizations
The optimization process focused primarily on the Pairformer architecture, which underpins state-of-the-art structure prediction models like AlphaFold3, OpenFold3, and Boltz-2. These models rely heavily on triangle attention and triangle multiplication, operations characterized by cubic scaling in both time and memory: , where represents the sequence length. To mitigate these bottlenecks, the researchers utilized Claude to develop FlashPairformer, a suite of custom GPU kernels. These kernels replace standard implementations with highly efficient, low-level software translation layers, yielding performance improvements of 1.7x to 3.2x in triangle operations.Beyond kernel development, Claude performed model-specific optimizations by:
- Redundant Computation Caching: Identifying and storing intermediate values to prevent re-computation.
- Constant Folding: Simplifying "dead branches" within the model architecture into constant outputs.
These efforts resulted in an average 4x speed-up across 30+ deep learning models—ranging from structure prediction and protein design to genomics—with minimal impact on precision.
Enabling Large-Scale Biomolecular Modeling
Claude introduced a low-memory "Big" mode, allowing for the inference of biomolecular systems previously requiring multi-node clusters to run on a single NVIDIA GPU. This mode facilitates the prediction of systems exceeding 10,000 tokens (amino acids/nucleotides/atoms), with successful proof-of-concept inference demonstrated on massive structures up to 70,000 tokens. This capability extends model utility into the domain of large molecular machines such as the human mitochondrial complex I, the TRiC chaperone, and the 70S ribosome.Advancements in De Novo Protein Design
The researchers demonstrated that Claude’s efficiency gains significantly lower the barrier to entry for drug discovery. By orchestrating optimized open-source design models, a single Claude instance—operating with a 1,100-word prompt and access to one NVIDIA H200 GPU—achieved *in silico* binding scores (*ipSAE*) comparable to previous, much more resource-intensive campaigns. The study reported that these results were achieved using roughly two orders of magnitude fewer GPU hours compared to previous agentic workflows, with total costs (compute and tokens) reduced to approximately $150 per target.Community Initiatives
To foster further innovation, the project is:- Open-Sourcing: Releasing all optimized code for the 30+ models to the broader scientific community.
- Protein Design Competition: Partnering with Adaptyv Bio, Modal, and Twist Bioscience to host a competition featuring five frontier-level design problems. The initiative includes up to 250,000 in Modal compute credits, and wet lab validation for over 5,000 protein designs.
- Life Sciences Verification Program: Expanding access to frontier AI capabilities for verified life science research organizations.