Skip to content

Commit 4ed16c8

Browse files
Merge pull request #3447 from AI-Hypercomputer:bvandermoon-docker-builds
PiperOrigin-RevId: 885810042
2 parents 093ab89 + 69198f7 commit 4ed16c8

1 file changed

Lines changed: 7 additions & 0 deletions

File tree

docs/install_maxtext.md

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@ MaxText offers three installation modes:
2222
1. maxtext[tpu]. Used for pre-training and decode on TPUs.
2323
2. maxtext[cuda12]. Used for pre-training and decode on GPUs.
2424
3. maxtext[tpu-post-train]. Used for post-training on TPUs. Currently, this option should also be used for running vllm_decode on TPUs.
25+
4. maxtext[runner]. Used for building MaxText's Docker images and scheduling workloads through XPK.
2526

2627
## From PyPI (Recommended)
2728

@@ -49,6 +50,9 @@ install_maxtext_cuda12_github_dep
4950
# Option 3: Installing maxtext[tpu-post-train]
5051
uv pip install "maxtext[tpu-post-train]>=0.2.0" --resolution=lowest
5152
install_maxtext_tpu_post_train_extra_deps
53+
54+
# Option 4: Installing maxtext[runner]
55+
uv pip install maxtext[runner] --resolution=lowest
5256
```
5357

5458
> **Note:** The `install_maxtext_tpu_github_deps`, `install_maxtext_cuda12_github_dep`, and
@@ -85,6 +89,9 @@ install_maxtext_cuda12_github_dep
8589
# Option 3: Installing .[tpu-post-train]
8690
uv pip install -e .[tpu-post-train] --resolution=lowest
8791
install_maxtext_tpu_post_train_extra_deps
92+
93+
# Option 4: Installing maxtext[runner]
94+
uv pip install .[runner] --resolution=lowest
8895
```
8996

9097
After installation, you can verify the package is available with `python3 -c "import maxtext"` and run training jobs with `python3 -m maxtext.trainers.pre_train.train ...`.

0 commit comments

Comments
 (0)