Neuron 2.26.1 is released! Check What's New and Announcements for more details.

Announcing end of support for jax_neuronx.nki_call API in jax-neuronx from starting next release

Announcing end of support for `jax_neuronx.nki_call` API in `jax-neuronx` from starting next release#

Starting with Neuron Release 2.23, Neuron will end support for jax_neuronx.nki_call API in jax-neuronx package.

For a full list of features that require jax-neuronx, please see JAX NeuronX Known Issues.

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