AWS Neuron SDK, Trainium용 NKI 커널 개발에 Neuron Agentic Development 공개

AWS Neuron이 Neuron Agentic Development를 공개해 Trainium과 Inferentia용 개발을 돕는다.

초기 릴리스는 Neuron Kernel Interface(NKI) 커널 개발에 맞춰져 있으며, NKI는 Trainium에 대한 저수준 프로그래밍 접근을 제공해 하드웨어 성능을 극대화하는 맞춤형 compute kernel 작성에 쓰인다.

개발자는 Claude Code나 Kiro 같은 agentic IDE에서 자연어로 작업을 지시할 수 있다. PyTorch 연산을 설명하면 NKI 커널을 받아보고, compilation error 수정이나 bottleneck 분석도 맡길 수 있다.

핵심 기능은 다음과 같다.

• kernel authoring
• debugging
• documentation lookup
• profile capture
• profile analysis

Neuron Agentic Development는 Neuron 스택 전반으로 확장될 open-source framework이며, 저장소는 GitHub에 공개됐다.