Tutorials

• 1. Simple NKIPy Tutorial
  • 1.1. Defining A NKIPy Kernel
  • 1.2. Running a NKIPy Kernel as a NumPy function
  • 1.3. Tracing a NKIPy Kernel
  • 1.4. Running the Traced Kernel with Simulation
  • 1.5. Running it On Trainium Hardware
• 2. Generate NKI Kernels from NKIPy Kernels
  • 2.1. Softmax Examples
  • 2.2. Matrix Multiplication Examples
• 3. MLP NKIPy Tutorial
  • 3.1. Defining A Simple MLP NKIPy Kernel
  • 3.2. Running the MLP Kernel as a NumPy function
  • 3.3. Tracing the MLP Kernel
  • 3.4. Running the Traced Kernel with Simulation
  • 3.5. Running it On Trainium Hardware