I can't wait to get access, sounds amazing. could you approve?

[TomNVIDIA]

Hello, JAX is now available.

ANNOUNCEMENT: General Availability of JAX NGC containers

We are excited to announce the GA release of the latest versions of JAX, Paxml, and T5x containers on NGC. NVIDIA NGC Catalog containers include a validated set of libraries optimized for GPU performance. Before each release, the containers are scanned for security vulnerabilities and performance across a range of popular DL models and workloads. These containers may also include experimental features in order to maximize performance and compatibility with NVIDIA GPUs. Highlights from 23.08 rel…