AWS Neuron SDK が Trainium での NKI カーネル開発用の Neuron Agentic Development で利用可能に
AWS Neuron では、AI コーディングアシスタントを備えた、エージェントとスキルのオープンソースコレクションである Neuron Agent Development 機能を発表しました。これにより、AWS Trainium および AWS Inferentia での開発をさらに加速させることができます。初回リリースでは、Neuron Kernel Interface (NKI) カーネル開発用のエージェントコーディング機能が提供されます。コードの作成からプロファイリング、パフォーマンス分析までのワークフローに対応します。
デベロッパーは、NKI を使用することで、ハードウェアのパフォーマンスを最大化するカスタム計算カーネルを作成するために、低レイヤーのプログラムで直接 Trainium にアクセスできます。Neuron Agentic Development では、デベロッパーのエージェント IDE (Claude Code や Kiro など) に NKI の専門知識を自然言語で直接取り込むことができます。例えば、デベロッパーは PyTorch の操作を説明して、動作している NKI カーネルを作成することや、エージェントにコンパイルエラーの修正を依頼し、問題を自動的に特定して修正を適用することができます。また、パフォーマンス分析をリクエストして、カーネルコードのどの行がボトルネックを引き起こしているかを特定するレポートを受け取ることもできます。これらの機能は、カーネルの作成、デバッグ、ドキュメントの参照、プロファイルの取得、プロファイル分析といった一連の工程を網羅しています。
Neuron Agentic Development は、Neuron スタック全体にわたってエージェント機能を活用するための包括的なフレームワークとして設計されており、その初期リリースとして NKI カーネル開発向けの機能が提供されました。リポジトリは GitHub で公開されています。
詳細はこちら: