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

Announcing end of support for torch_neuronx.nki_jit API in torch-neuronx starting next release

Neuron Release 2.23 will be the last release to include support for torch_neuronx.nki_jit API in torch-neuronx package.

Customers using torch_neuronx.nki_jit API are recommended to switch invocations to directly call functions annotated with nki.jit API.

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