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