NKI Deep Dives#

This section provides in-depth technical documentation and guides for advanced users of the Neuron Kernel Interface (NKI). These deep dives offer detailed explanations of NKI concepts, programming patterns, and best practices to help you maximize the performance and capabilities of your NKI code on AWS Neuron devices.

Advanced NKI Programming#

NKI Language Guide (Beta 2)

Comprehensive guide to the NKI language, including syntax, tensor operations, control flow, and class support for Trainium devices.

NKI Programming Model (Legacy)

Core concepts of NKI programming including memory hierarchy, tiling, and execution model.