.. _nki_tutorials:

NKI Tutorials
================================

The full source code of the following tutorials can be also viewed on the 
`nki-samples <https://github.com/aws-neuron/nki-samples/tree/main/src/nki_samples/tutorials/>`_ repository on GitHub.

.. toctree::
      :maxdepth: 1
      :hidden:

      tutorials/spmd_tensor_addition
      tutorials/spmd_multiple_nc_tensor_addition
      tutorials/transpose2d
      tutorials/average_pool2d
      tutorials/matrix_multiplication
      tutorials/rmsnorm
      tutorials/layernorm
      tutorials/fused-self-attn
      tutorials/fused_mamba



* :doc:`SPMD Tensor addition <tutorials/spmd_tensor_addition>`
* :doc:`SPMD Tensor addition with multiple Neuron Cores <tutorials/spmd_multiple_nc_tensor_addition>`
* :doc:`Transpose2D <tutorials/transpose2d>`
* :doc:`AveragePool2D <tutorials/average_pool2d>`
* :doc:`Matrix multiplication <tutorials/matrix_multiplication>`
* :doc:`RMSNorm <tutorials/rmsnorm>`
* :doc:`Fused Self Attention <tutorials/fused-self-attn>`
* :doc:`LayerNorm <tutorials/layernorm>`
* :doc:`Fused Mamba <tutorials/fused_mamba>`