NKI-Agent: Domain-Specific Fine-Tuning and Agentic Tool Use for Neuron Kernel Generation
NKI-Agent:用于神经元内核生成的特定领域微调与智能体工具使用
机构 * AWS Neuron Team(亚马逊云科技神经元团队) ; Amazon Web Services(亚马逊云科技)
专题命中 工具调用 :agent(title,title_cn);tool use(title,abstract);agentic(title,abstract);分类 cs.LG
AI总结 针对新兴AI加速器,介绍NKI-Agent系统,结合特定领域监督微调与编译验证修复智能体循环生成NKI内核,适配框架、构建基准并评估,显示工具使用关键及不同模型表现,为强化学习奖励设计提供指导。
Comments 7 pages. Accepted at DL4C @ ICML 2026 Workshop