For AI agents: a documentation index is available at the root level at /llms.txt. Append /llms.txt to any URL for a page-level index, or .md for the markdown version of any page.
LogoLogoDocumentation
  • Getting Started
    • Introduction
    • JAX on NVIDIA GPU Stack
    • Frameworks & Supported Models
    • Build Pipeline Status
    • Environment Variables
  • Frameworks
    • AXLearn
  • Performance & Profiling
    • Profiling
    • GPU Performance
    • PGLE (Profile-Guided Latency Estimation)
    • Native FP8
    • nsys-jax
  • Resiliency
    • Resilient JAX with Ray
    • Example Walkthrough
  • Inference
    • JAX-vLLM Rollout Offloading Bridge
  • Customization
    • GPU Kernels
  • Reference
    • Container Versions
    • Staging Containers
    • Cloud Providers
    • FAQ
    • Resources & Videos
  • Introduction
  • JAX on NVIDIA GPU Stack
  • Frameworks & Supported Models
  • Build Pipeline Status
  • Environment Variables
  • Overview
  • NVFP4 Training Example
  • AXLearn
  • Profiling
  • GPU Performance
  • PGLE (Profile-Guided Latency Estimation)
  • Native FP8
  • nsys-jax
  • Resilient JAX with Ray
  • Example Walkthrough
  • JAX-vLLM Rollout Offloading Bridge
  • GPU Kernels
  • Container Versions
  • Staging Containers
  • Cloud Providers
  • FAQ
  • Resources & Videos
On this page
  • AWS
  • GCP
  • Azure
  • OCI
Reference

Cloud Providers

Running JAX on public clouds
||View as Markdown|

AWS

  • Add EFA integration
  • SageMaker code sample

GCP

  • Getting started with JAX multi-node applications with NVIDIA GPUs on Google Kubernetes Engine

Azure

  • Accelerating AI applications using the JAX framework on Azure’s NDm A100 v4 Virtual Machines

OCI

  • Running a deep learning workload with JAX on multinode multi-GPU clusters on OCI
Previous

Staging Containers

Next

FAQ

NVIDIANVIDIA
Developer-friendly docs for your API
Privacy Policy | Your Privacy Choices | Terms of Service | Accessibility | Corporate Policies | Product Security | Contact

Copyright © 2026, NVIDIA Corporation.