NVIDIA BioNeMo Agent Toolkit 加速端到端共折叠流程
NVIDIA Generativ17 天前
生物分子结构预测与共折叠任务，正在成为药物发现和蛋白质设计中的主流大规模工作负载。以 OpenFold3 这类模型为代表的系统，不再只是单独完成结构预测，而是逐步嵌入由 AI Agent 驱动的端到端研发流程。
在这样的流程中，性能瓶颈并不只出现在模型推理阶段。一个完整的共折叠工作流通常需要覆盖多个环节：
- 多序列比对生成
- 共折叠推理
- 模型服务化部署
- 多 GPU 横向扩展
如果希望 AI Agent 稳定、高效地运行整个流程，每个环节都需要具备足够的速度与可扩展性。否则，即使核心模型本身表现良好，整体工作流仍可能受到前处理、服务吞吐或多卡调度效率的限制。
NVIDIA BioNeMo Agent Toolkit 的关注点，正是将这些步骤串联为更高效的端到端共折叠流程，使结构预测和蛋白质设计相关任务能够更适合大规模运行。