How-to guides#
Explore our how-to guides for optimizing, debugging, and managing your MaxText workloads.
⚡ Optimization
Techniques for maximizing performance, including sharding strategies, Pallas kernels, and benchmarking.
💾 Data Pipelines
Configure input pipelines using Grain (recommended for determinism), HuggingFace, or TFDS.
🔄 Checkpointing
Manage GCS checkpoints, handle preemption with emergency checkpointing, and configure multi-tier storage.
🔍 Monitoring & Debugging
Tools for observability: goodput monitoring, hung job debugging, and Vertex AI TensorBoard integration.
🐍 Python Notebooks
Interactive development guides for running MaxText on Google Colab or local JupyterLab environments.
🌱 Model Bringup
A step-by-step guide for the community to help expand MaxText’s model library.