This document is relevant for: Inf1, Inf2, Trn1, Trn2

AWS Neuron SDK 2.25.0: Neuron Kernel Interace (NKI) release notes#

Date of release: July 31, 2025

Behavioral changes#

Behavioral changes are small, user-facing changes that you may notice after upgrading to this version.

  • Added a new NKI ISA select_reduce instruction that combines selection and reduction operations in a single instruction. For more details, see the NKI API reference.

  • Improved nl.store ergonomics to perform a copy to sbuf prior to the store to HBM if the value tile is in PSUM. For more details, see nl_store.

  • Added support for dict, list, and dataclass arguments containing tensors as inputs to NKI kernels.

This document is relevant for: Inf1, Inf2, Trn1, Trn2