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

AWS Neuron SDK 2.25.0: JAX support release notes

Date of release: July 31, 2025

• Go back to the AWS Neuron 2.25.0 release notes home

Released versions

• 0.6.1.1.0.*

Improvements

• This release introduces support for JAX version 0.6.1.

Bug fixes

• Previously, using multiple meshes within a single program wasn’t supported. This is fixed to add support for sub-meshes.

Known issues

• Known issues are listed at JAX NeuronX Known Issues.

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