From 25183c0ffbd53f031c9100bc1e5585872e5634aa Mon Sep 17 00:00:00 2001 From: Branden Vandermoon Date: Wed, 25 Mar 2026 17:25:04 +0000 Subject: [PATCH] Specifically install maxtext 0.2.1 in documentation --- docs/build_maxtext.md | 2 +- docs/install_maxtext.md | 8 ++++---- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/docs/build_maxtext.md b/docs/build_maxtext.md index 83cd058555..95191ea592 100644 --- a/docs/build_maxtext.md +++ b/docs/build_maxtext.md @@ -61,7 +61,7 @@ source ${VENV_NAME?}/bin/activate # Install MaxText with the [runner] extra # This enables Docker image building and workload scheduling via XPK -uv pip install maxtext[runner] --resolution=lowest +uv pip install maxtext[runner]==0.2.1 --resolution=lowest ``` > **Note:** The `maxtext[runner]` extra includes all necessary dependencies for building MaxText Docker images and running workloads through XPK. It automatically installs XPK, so you do not need to install it separately to manage your clusters and workloads. diff --git a/docs/install_maxtext.md b/docs/install_maxtext.md index 05f720b104..5580dd68e6 100644 --- a/docs/install_maxtext.md +++ b/docs/install_maxtext.md @@ -40,19 +40,19 @@ source maxtext_venv/bin/activate # installation option from this list to fit your use case. # Option 1: Installing maxtext[tpu] -uv pip install maxtext[tpu] --resolution=lowest +uv pip install maxtext[tpu]==0.2.1 --resolution=lowest install_maxtext_tpu_github_deps # Option 2: Installing maxtext[cuda12] -uv pip install maxtext[cuda12] --resolution=lowest +uv pip install maxtext[cuda12]==0.2.1 --resolution=lowest install_maxtext_cuda12_github_dep # Option 3: Installing maxtext[tpu-post-train] -uv pip install maxtext[tpu-post-train] --resolution=lowest +uv pip install maxtext[tpu-post-train]==0.2.1 --resolution=lowest install_maxtext_tpu_post_train_extra_deps # Option 4: Installing maxtext[runner] -uv pip install maxtext[runner] --resolution=lowest +uv pip install maxtext[runner]==0.2.1 --resolution=lowest ``` > **Note:** The `install_maxtext_tpu_github_deps`, `install_maxtext_cuda12_github_dep`, and