NVIDIA Docs Hub Homepage NVIDIA Optimized Frameworks NVIDIA Optimized Frameworks JAX Release Notes

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 on NGC four times a year to provide you with the latest NVIDIA deep learning software libraries and GitHub code contributions that have been sent upstream.

Nightly builds of the containers are available on JAX Toolbox. 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.

Note:

Security Common Vulnerabilities and Exposures (CVEs)

Please review the Security Scanning tab on NGC to see the latest security scan results. For certain open-source vulnerabilities listed in the scan results, NVIDIA provides a response in the form of a Vulnerability Exploitability eXchange (VEX) document. The VEX information can be reviewed and downloaded from the Security Scanning tab.

JAX Release Notes

JAX Release Notes

Table of Contents