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

Neuron no longer supports jax_neuronx.nki_call API in jax-neuronx starting this release

Neuron no longer supports `jax_neuronx.nki_call` API in `jax-neuronx` starting this release#

Neuron Release 2.23 no longer supports 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 will need to switch invocations to directly call functions annotated with nki.jit API.