Run MaxText#
Choose your environment and orchestration method to run MaxText.
💻 Localhost / Single VM
Get started quickly on a single machine. Clone the repo, install dependencies, and run your first training job on a single TPU or GPU VM.
🎮 Single-host GPU
Run MaxText on single-host NVIDIA GPUs (e.g., A3 High/Mega). Includes Docker setup, NVIDIA Container Toolkit installation, and 1B/7B model training examples.
🏗️ At scale with XPK (GKE)
Deploy to Google Kubernetes Engine (GKE) using XPK. Orchestrate large-scale training jobs on TPU or GPU clusters with simple CLI commands.
🌐 Multi-host via Pathways
Run large-scale JAX jobs on TPUs using Pathways. Supports batch and headless (interactive) workloads on GKE.
🔌 Decoupled Mode
Run tests and local development without Google Cloud dependencies (no gcloud, GCS, or Vertex AI required).