AWS Neuron SDK, 이제 Trainium에서 NKI 커널 개발을 위해 Neuron Agentic Development과 함께 사용 가능
AWS Neuron은 AI 코딩 어시스턴트가 AWS Trainium 및 AWS Inferentia에서 개발을 가속화할 수 있도록 지원하는 오픈 소스 에이전트 및 기술 모음인 Neuron Agentic Development 기능을 발표했습니다. 초기 릴리스에서는 작성부터 프로파일링, 성능 분석까지의 워크플로를 포괄하는 Neuron Kernel Interface(NKI) 커널 개발을 위한 에이전트 코딩 기능을 제공합니다.
NKI는 개발자가 하드웨어 성능을 극대화하는 사용자 지정 컴퓨팅 커널을 작성할 수 있도록 Trainium에 대한 직접적인 저수준 프로그래밍 액세스를 제공합니다. Neuron Agentic Development는 자연어를 통해 NKI 전문 지식을 개발자의 에이전트 IDE(예: Claude Code 및 Kiro)에 직접 제공합니다. 예를 들어 개발자는 PyTorch 작업을 설명하고 작동하는 NKI 커널을 받거나, 에이전트에게 컴파일 오류를 수정하고 자동으로 문제를 식별하여 수정 사항을 적용하도록 요청하거나, 성능 분석을 요청하여 병목 현상을 일으키는 커널 코드 줄을 식별하는 보고서를 받을 수 있습니다. 기능에는 커널 작성, 디버깅, 문서 조회, 프로파일 캡처, 프로파일 분석이 포함됩니다.
Neuron Agentic Development는 Neuron 스택 전반에 걸쳐 에이전트 기능을 위한 광범위한 프레임워크로 설계되었으며, NKI 커널 개발이 초기 릴리스입니다. 리포지토리는 GitHub에서 사용할 수 있습니다.
자세히 알아보기: