Approval Pending

[c.schesch]

Hi, I’d like to try Jax for high-performance computing on Nvidia GPUs on a university cluster. I applied for preview, let me know if you guys can grant access. Thanks a lot in advance! :)

[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…