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

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#

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