JAX Release Notes

JAX Release Notes (PDF)

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.

© Copyright 2023, NVIDIA. Last updated on Feb 29, 2024.