AWS Neuron SDK 现已支持 Neuron Agentic Development,用于在 Trainium 上进行 NKI 内核开发

发布于: 2026年4月30日

AWS Neuron 宣布推出 Neuron Agentic Development 功能。这是一套开源的代理和技能集合,旨在为人工智能编码助手提供支持,从而加速在 AWS TrainiumAWS Inferentia 上的开发流程。此初始版本为 Neuron Kernel 接口(NKI)内核开发提供了代理式编码能力,涵盖从编写到性能剖析及分析的整个工作流。

借助 NKI,开发人员能够直接以低级编程方式访问 Trainium,编写自定义计算内核,从而最大限度发挥硬件性能。Neuron Agentic Development 通过自然语言将 NKI 专业知识直接融入开发人员的代理式 IDE(例如 Claude Code 和 Kiro)。例如,开发人员可以描述 PyTorch 操作,然后获得可正常运行的 NKI 内核;要求代理修复编译错误,使其自动识别问题并进行修正;或者请求性能分析并收到一份报告,指出内核代码中的哪些行导致了瓶颈。其功能涵盖内核编写、调试、文档查询、性能分析数据采集以及性能分析。

Neuron Agentic Development 涉及用作覆盖整个 Neuron 堆栈中代理能力的通用框架,NKI 内核开发是其首个发布版本。该代码库已在 GitHub 上提供。

了解详情: