NVIDIA BioNeMo Inference Runtime Delivers 2.9x Faster Structure Prediction for Boltz-2

PyTorch · x · 2026-09-18

NVIDIA's new technical blog details BioNeMo Inference Runtime (BioIR), which accelerates biomolecular structure-prediction models while keeping the PyTorch workflow.

Key benchmark (1,000 human dimers, 8xH100):

Three optimization layers: kernel selection, CUDA Graph capture for module optimization, and Ray replicas for pipeline scaling (full model replica per GPU, overlapping CPU stages with GPU folding).

Supports Boltz-2, OpenFold2, and OpenFold3 end-to-end (parsing, tokenization, feature generation, GPU inference, PDB/mmCIF output), and integrates as torch.nn.Module for custom code.

Original post →

More from Infra

Infra channel →