在 SageMaker JumpStart 上使用 NVIDIA NIM 构建可随时投入生产的药物研发和机器人管道
Amazon SageMaker JumpStart 现支持一键部署四款专为生物科学和物理人工智能打造的 NVIDIA NIM 模型:ProteinMPNN、Nemotron-3.5B-Instruct、MSA Search NIM 和 Cosmos Reason。NVIDIA NIM™ 提供预先构建、优化的推理微服务,用于在任何 NVIDIA 加速基础设施上快速部署最新的人工智能模型。这些模型具备先进能力,涵盖蛋白质设计、可配置输出推理以及物理世界理解,助力客户在 AWS 基础设施上加速推进生物科学研究、药物发现及具身人工智能应用。
ProteinmPNN 可在结构数据的指导下实现快速高效的蛋白质序列优化。该 NIM 可生成具有增强结合亲和力和稳定性的高质量序列,并通过实验结果得到验证。ProteinMPNN 专为可扩展性和灵活性而设计,可无缝集成到蛋白质工程工作流程中,彻底改变酶设计和治疗药物开发等应用领域。
MSA Search NIM 支持针对查询氨基酸序列与蛋白质序列数据库集进行 GPU 加速的多序列比对(MSA)。这些数据库会搜索与查询序列相似的序列,然后将这些序列集合进行比对,从而确定相似区域,即使蛋白质的长度和基序不同。
Nemotron-3.5B-Instruct 提供卓越的推理性能、原生工具调用支持,以及基于 256k 令牌上下文窗口的扩展上下文处理能力。该模型采用高效的混合专家(MoE)架构,可在代理式工作负载与编码工作负载中实现比前代模型更高的吞吐量,同时保持大型模型的深度推理能力。 它非常适合构建多代理工作流程、开发人员生产力工具、流程自动化以及科学和数学推理分析等。
Cosmos Reason 是一款面向物理人工智能与机器人技术的开放式、可定制推理视觉语言模型(VLM)。它使机器人和视觉人工智能代理能够像人类一样推理,利用先验知识、物理理解和常识来理解现实世界并采取行动。该模型了解空间、时间和基础物理学,可以作为规划模型来推理具身代理下一步可能采取的步骤。
借助 SageMaker JumpStart,客户只需点击几下即可部署这些模型中的任何一个,以解决其特定的人工智能使用案例。
要开始使用这些模型,请导航到 SageMaker 控制台中的 SageMaker JumpStart 模型目录,或使用 SageMaker Python SDK 将模型部署到您的 AWS 账户。有关在 SageMaker JumpStart 中部署和使用基础模型的更多信息,请参阅 Amazon SageMaker JumpStart 文档。