This document is relevant for: Trn1, Trn2, Trn3

NKI Migration Guides

These guides help you migrate your NKI kernels between versions.

NKI 0.3.0 Update Guide

Migrate your NKI kernels from 0.2.0 to 0.3.0, including API changes, deprecations, and new features.

NKI Block Dimension Migration Guide

Migrate NKI kernels to use block dimensions for improved performance and resource utilization on Trainium devices.

NKI Beta 2 Migration Guide

Migrate NKI kernels from Beta 1 to Beta 2.

This document is relevant for: Trn1, Trn2, Trn3