NVIDIA BioNeMo 推理运行时:蛋白质结构预测吞吐量提升 2.9 倍
AllThingsApx · x · 2026-09-11
NVIDIA 技术博客介绍 BioNeMo Inference Runtime(BioIR),用于在 GPU 上加速生物分子结构预测模型,同时保留熟悉的 PyTorch 工作流。
- 端到端处理器:覆盖输入解析、tokenization、特征生成、GPU 推理到 PDB/mmCIF 输出,也支持直接集成 PyTorch 模块用于自定义代码。
- 高吞吐设计:Ray executor 在单节点每块 GPU 上放置完整模型副本,通过 CPU 阶段与 GPU 折叠重叠提升吞吐。
- 实测数据:在 8xH100 上对 1000 个人类二聚体目标的基准测试中,BioIR 加速的 Boltz-2 达到每 GPU 小时 5.85 万残基,对比 torch-compiled 开源实现的 2.02 万,吞吐提升 2.90 倍。
- 三层优化:kernel 选择、CUDA Graph 捕获的模块优化、Ray 副本的流水线扩展,分别针对不同瓶颈。
- 能耗:折叠一百万个同类目标的估计能耗为 11 MWh,对比公开实现的 35 MWh。
所属事件:NVIDIA 开源 BioNeMo 推理运行时,蛋白结构预测提速近 3 倍(19 条相关)→
「Infra」频道最新
- Anthropic 营收三年暴涨 4000 倍,上半年已入账 163 亿美元 — daniel_mac8 · 2026-10-27
- Tenstorrent 芯片跑通蛋白质生成模型,宣称无 HBM 短缺之忧 — gajesh · 2026-10-27
- YC 系创企 Hopper 出击语音推理:TTFT 80ms、价格为 GPT-4.1 的 15% — ycombinator · 2026-10-27
- a16z 数据:A100 租赁价格 2026 年仍在涨,旧卡并未贬值 — VraserX · 2026-10-27
- Anthropic S-1 曝超 8000 亿美元未入表的未来算力承诺 — GaryMarcus · 2026-10-27
- Anthropic 约 11GW 算力合同中 NVIDIA GPU 占约 42% — ns123abc · 2026-10-27