JAX Release Notes
This document describes the key features, software enhancements and improvements, known issues, and how to run this container. The JAX framework allows you to develop deep learning models with flexibility and performance. The JAX container is released several times a year to provide you with the latest NVIDIA deep learning software libraries and GitHub code contributions that have been sent upstream. The libraries and contributions have all been tested, tuned, and optimized.
For a complete view of the supported software and specific versions that are packaged with the frameworks based on the container image, see the Frameworks Support Matrix.