NKI Tutorials

The full source code of the following tutorials can be also viewed on the nki-samples repository on GitHub.

• SPMD Tensor addition

• SPMD Tensor addition with multiple Neuron Cores

• Transpose2D

• AveragePool2D

• Matrix multiplication

• RMSNorm

• Fused Self Attention

• LayerNorm

• Fused Mamba