Skip to content

Commit 30866db

Browse files
bvandermoonbvandermoon
authored
Merge pull request #69 from AI-Hypercomputer/bvandermoon-tpu-recipes
Point to correct step number when builing MaxText Docker image
2 parents c4e6b90 + c2ba6d8 commit 30866db

File tree

7 files changed

+21
-21
lines changed

7 files changed

+21
-21
lines changed

‎training/trillium/GPT3-175B-MaxText/bf16/README.md‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,19 +1,19 @@
11
# Instructions for training GPT3-175B-Maxtext on TPU trillium
22

33
## XPK setup
4-
Please follow this [link](https://github.com/AI-Hypercomputer/tpu-recipes/blob/main/training/trillium/XPK_README.md) to create your GKE cluster with XPK
4+
Please follow the [XPK_README](https://github.com/AI-Hypercomputer/tpu-recipes/blob/main/training/trillium/XPK_README.md) to create your GKE cluster with XPK
55

66
## Prep for Maxtext
77

88
### Install MaxText and Build Docker Image
9-
Please follow this [link](https://github.com/AI-Hypercomputer/tpu-recipes/blob/main/training/trillium/MAXTEXT_README.md) to install maxtext and build the docker image. The following variables should be set:
9+
Please follow the [MAXTEXT_README](https://github.com/AI-Hypercomputer/tpu-recipes/blob/main/training/trillium/MAXTEXT_README.md) to install maxtext and build the docker image. The following variables should be set:
1010

1111
In step 1, use the MaxText [tpu-recipes-v0.1.2](https://github.com/AI-Hypercomputer/maxtext/releases/tag/tpu-recipes-v0.1.2) tag to run this recipe:
1212
```
1313
git checkout tpu-recipes-v0.1.2
1414
```
1515

16-
In step 2, use the jax-stable-stack image containing JAX 0.5.2:
16+
In step 3, use the jax-stable-stack image containing JAX 0.5.2:
1717
```
1818
BASE_IMAGE=us-docker.pkg.dev/cloud-tpu-images/jax-stable-stack/tpu:jax0.5.2-rev1
1919
bash docker_build_dependency_image.sh DEVICE=tpu MODE=stable_stack BASEIMAGE=${BASE_IMAGE}

‎training/trillium/Llama2-70B-MaxText/README.md‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,19 +1,19 @@
11
# Instructions for training Llama2-70B-Maxtext on TPU trillium
22

33
## XPK setup
4-
Please follow this [link](https://github.com/AI-Hypercomputer/tpu-recipes/blob/main/training/trillium/XPK_README.md) to create your GKE cluster with XPK
4+
Please follow the [XPK_README](https://github.com/AI-Hypercomputer/tpu-recipes/blob/main/training/trillium/XPK_README.md) to create your GKE cluster with XPK
55

66
## Prep for Maxtext
77

88
### Install MaxText and Build Docker Image
9-
Please follow this [link](https://github.com/AI-Hypercomputer/tpu-recipes/blob/main/training/trillium/MAXTEXT_README.md) to install maxtext and build the docker image. The following variables should be set:
9+
Please follow the [MAXTEXT_README](https://github.com/AI-Hypercomputer/tpu-recipes/blob/main/training/trillium/MAXTEXT_README.md) to install maxtext and build the docker image. The following variables should be set:
1010

1111
In step 1, use the MaxText [tpu-recipes-v0.1.2(https://github.com/AI-Hypercomputer/maxtext/releases/tag/tpu-recipes-v0.1.2) tag to run this recipe:
1212
```
1313
git checkout tpu-recipes-v0.1.2
1414
```
1515

16-
In step 2, use the jax-stable-stack image containing JAX 0.5.2:
16+
In step 3, use the jax-stable-stack image containing JAX 0.5.2:
1717
```
1818
BASE_IMAGE=us-docker.pkg.dev/cloud-tpu-images/jax-stable-stack/tpu:jax0.5.2-rev1
1919
bash docker_build_dependency_image.sh DEVICE=tpu MODE=stable_stack BASEIMAGE=${BASE_IMAGE}

‎training/trillium/Llama3-8B-MaxText/v6e-8/README.md‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,19 +1,19 @@
11
# Instructions for training Llama3.1-8B-MaxText on TPU trillium (v6e-8)
22

33
## XPK setup
4-
Please follow this [link](https://github.com/AI-Hypercomputer/tpu-recipes/blob/main/training/trillium/XPK_README.md) to create your GKE cluster with XPK
4+
Please follow the [XPK_README](https://github.com/AI-Hypercomputer/tpu-recipes/blob/main/training/trillium/XPK_README.md) to create your GKE cluster with XPK
55

66
## Prep for Maxtext
77

88
### Install MaxText and Build Docker Image
9-
Please follow this [link](https://github.com/AI-Hypercomputer/tpu-recipes/blob/main/training/trillium/MAXTEXT_README.md) to install maxtext and build the docker image. The following variables should be set:
9+
Please follow the [MAXTEXT_README](https://github.com/AI-Hypercomputer/tpu-recipes/blob/main/training/trillium/MAXTEXT_README.md) to install maxtext and build the docker image. The following variables should be set:
1010

1111
In step 1, use the MaxText [tpu-recipes-v0.1.2](https://github.com/AI-Hypercomputer/maxtext/releases/tag/tpu-recipes-v0.1.2) tag to run this recipe:
1212
```
1313
git checkout tpu-recipes-v0.1.2
1414
```
1515

16-
In step 2, use the jax-stable-stack image containing JAX 0.5.2:
16+
In step 3, use the jax-stable-stack image containing JAX 0.5.2:
1717
```
1818
BASE_IMAGE=us-docker.pkg.dev/cloud-tpu-images/jax-stable-stack/tpu:jax0.5.2-rev1
1919
bash docker_build_dependency_image.sh DEVICE=tpu MODE=stable_stack BASEIMAGE=${BASE_IMAGE}

‎training/trillium/Llama3.1-405B-MaxText/README.md‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,19 +1,19 @@
11
# Instructions for training Llama3.1-405B-MaxText on TPU trillium
22

33
## XPK setup
4-
Please follow this [link](https://github.com/AI-Hypercomputer/tpu-recipes/blob/main/training/trillium/XPK_README.md) to create your GKE cluster with XPK
4+
Please follow the [XPK_README](https://github.com/AI-Hypercomputer/tpu-recipes/blob/main/training/trillium/XPK_README.md) to create your GKE cluster with XPK
55

66
## Prep for Maxtext
77

88
### Install MaxText and Build Docker Image
9-
Please follow this [link](https://github.com/AI-Hypercomputer/tpu-recipes/blob/main/training/trillium/MAXTEXT_README.md) to install maxtext and build the docker image. The following variables should be set:
9+
Please follow the [MAXTEXT_README](https://github.com/AI-Hypercomputer/tpu-recipes/blob/main/training/trillium/MAXTEXT_README.md) to install maxtext and build the docker image. The following variables should be set:
1010

1111
In step 1, use the MaxText [tpu-recipes-v0.1.2](https://github.com/AI-Hypercomputer/maxtext/releases/tag/tpu-recipes-v0.1.2) tag to run this recipe:
1212
```
1313
git checkout tpu-recipes-v0.1.2
1414
```
1515

16-
In step 2, use the jax-stable-stack image containing JAX 0.5.2:
16+
In step 3, use the jax-stable-stack image containing JAX 0.5.2:
1717
```
1818
BASE_IMAGE=us-docker.pkg.dev/cloud-tpu-images/jax-stable-stack/tpu:jax0.5.2-rev1
1919
bash docker_build_dependency_image.sh DEVICE=tpu MODE=stable_stack BASEIMAGE=${BASE_IMAGE}

‎training/trillium/Llama3.1-70B-MaxText/README.md‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,19 +1,19 @@
11
# Instructions for training Llama3.1-70B-MaxText on TPU trillium (v6e-256)
22

33
## XPK setup
4-
Please follow this [link](https://github.com/AI-Hypercomputer/tpu-recipes/blob/main/training/trillium/XPK_README.md) to create your GKE cluster with XPK
4+
Please follow the [XPK_README](https://github.com/AI-Hypercomputer/tpu-recipes/blob/main/training/trillium/XPK_README.md) to create your GKE cluster with XPK
55

66
## Prep for Maxtext
77

88
### Install MaxText and Build Docker Image
9-
Please follow this [link](https://github.com/AI-Hypercomputer/tpu-recipes/blob/main/training/trillium/MAXTEXT_README.md) to install maxtext and build the docker image. The following variables should be set:
9+
Please follow the [MAXTEXT_README](https://github.com/AI-Hypercomputer/tpu-recipes/blob/main/training/trillium/MAXTEXT_README.md) to install maxtext and build the docker image. The following variables should be set:
1010

1111
In step 1, use the MaxText [tpu-recipes-v0.1.2](https://github.com/AI-Hypercomputer/maxtext/releases/tag/tpu-recipes-v0.1.2) tag to run this recipe:
1212
```
1313
git checkout tpu-recipes-v0.1.2
1414
```
1515

16-
In step 2, use the jax-stable-stack image containing JAX 0.5.2:
16+
In step 3, use the jax-stable-stack image containing JAX 0.5.2:
1717
```
1818
BASE_IMAGE=us-docker.pkg.dev/cloud-tpu-images/jax-stable-stack/tpu:jax0.5.2-rev1
1919
bash docker_build_dependency_image.sh DEVICE=tpu MODE=stable_stack BASEIMAGE=${BASE_IMAGE}

‎training/trillium/Mistral-7B-MaxText/README.md‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,19 +1,19 @@
11
# Instructions for training Mistral-7B-MaxText on TPU trillium (v6e-8)
22

33
## XPK setup
4-
Please follow this [link](https://github.com/AI-Hypercomputer/tpu-recipes/blob/main/training/trillium/XPK_README.md) to create your GKE cluster with XPK
4+
Please follow the [XPK_README](https://github.com/AI-Hypercomputer/tpu-recipes/blob/main/training/trillium/XPK_README.md) to create your GKE cluster with XPK
55

66
## Prep for Maxtext
77

88
### Install MaxText and Build Docker Image
9-
Please follow this [link](https://github.com/AI-Hypercomputer/tpu-recipes/blob/main/training/trillium/MAXTEXT_README.md) to install maxtext and build the docker image. The following variables should be set:
9+
Please follow the [MAXTEXT_README](https://github.com/AI-Hypercomputer/tpu-recipes/blob/main/training/trillium/MAXTEXT_README.md) to install maxtext and build the docker image. The following variables should be set:
1010

1111
In step 1, use the MaxText [tpu-recipes-v0.1.2](https://github.com/AI-Hypercomputer/maxtext/releases/tag/tpu-recipes-v0.1.2) tag to run this recipe:
1212
```
1313
git checkout tpu-recipes-v0.1.2
1414
```
1515

16-
In step 2, use the jax-stable-stack image containing JAX 0.5.2:
16+
In step 3, use the jax-stable-stack image containing JAX 0.5.2:
1717
```
1818
BASE_IMAGE=us-docker.pkg.dev/cloud-tpu-images/jax-stable-stack/tpu:jax0.5.2-rev1
1919
bash docker_build_dependency_image.sh DEVICE=tpu MODE=stable_stack BASEIMAGE=${BASE_IMAGE}

‎training/trillium/Mixtral-8x7B-MaxText/README.md‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,19 +1,19 @@
11
# Instructions for training Mixtral-8x7B-MaxText on TPU trillium
22

33
## XPK setup
4-
Please follow this [link](https://github.com/AI-Hypercomputer/tpu-recipes/blob/main/training/trillium/XPK_README.md) to create your GKE cluster with XPK
4+
Please follow the [XPK_README](https://github.com/AI-Hypercomputer/tpu-recipes/blob/main/training/trillium/XPK_README.md) to create your GKE cluster with XPK
55

66
## Prep for Maxtext
77

88
### Install MaxText and Build Docker Image
9-
Please follow this [link](https://github.com/AI-Hypercomputer/tpu-recipes/blob/main/training/trillium/MAXTEXT_README.md) to install maxtext and build the docker image. The following variables should be set:
9+
Please follow the [MAXTEXT_README](https://github.com/AI-Hypercomputer/tpu-recipes/blob/main/training/trillium/MAXTEXT_README.md) to install maxtext and build the docker image. The following variables should be set:
1010

1111
In step 1, use the MaxText [tpu-recipes-v0.1.2](https://github.com/AI-Hypercomputer/maxtext/releases/tag/tpu-recipes-v0.1.2) tag to run this recipe:
1212
```
1313
git checkout tpu-recipes-v0.1.2
1414
```
1515

16-
In step 2, use the jax-stable-stack image containing JAX 0.5.2:
16+
In step 3, use the jax-stable-stack image containing JAX 0.5.2:
1717
```
1818
BASE_IMAGE=us-docker.pkg.dev/cloud-tpu-images/jax-stable-stack/tpu:jax0.5.2-rev1
1919
bash docker_build_dependency_image.sh DEVICE=tpu MODE=stable_stack BASEIMAGE=${BASE_IMAGE}

0 commit comments

Comments
 (0)