Refactor: Replace placeholder GCS bucket name (#187)

Replaced the placeholder GCS bucket name "prefix-artifact-repository"
with "your-prefix-artifact-repository" in all relevant files.

This change affects configuration files and example scripts for TPU
training on GKE.

Co-authored-by: google-labs-jules[bot] <161369871+google-labs-jules[bot]@users.noreply.github.com>

Author: mmontan

ai-infrastructure/tpu-training-on-gke/examples/jobset/maxtext/parameters.env.multi_slice_8B

@@ -5,8 +5,8 @@ ICI_PARALLELISM=16
 JOB_PARALLELISM=4
 NUM_SLICES=2
 RUN_NAME=maxtext-single-slice-101
-BASE_OUTPUT_DIRECTORY=gs://prefix-artifact-repository/runs
-DATASET_PATH=gs://prefix-artifact-repository/datasets
+BASE_OUTPUT_DIRECTORY=gs://your-prefix-artifact-repository/runs
+DATASET_PATH=gs://your-prefix-artifact-repository/datasets
 TENSORBOARD_NAME=projects/project-id/locations/us-central1/tensorboards/910xxxxx16
 WID_KSA=wid-sa
 ARGS=steps=150 per_device_batch_size=6 enable_checkpointing=false enable_profiler=false remat_policy=full max_target_length=2048  log_period=50 global_parameter_scale=8

ai-infrastructure/tpu-training-on-gke/examples/jobset/maxtext/parameters.env.single_slice_8B

@@ -5,8 +5,8 @@ ICI_PARALLELISM=16
 JOB_PARALLELISM=4
 NUM_SLICES=1
 RUN_NAME=maxtext-single-slice-101
-BASE_OUTPUT_DIRECTORY=gs://prefix-artifact-repository/runs
-DATASET_PATH=gs://prefix-artifact-repository/datasets
+BASE_OUTPUT_DIRECTORY=gs://your-prefix-artifact-repository/runs
+DATASET_PATH=gs://your-prefix-artifact-repository/datasets
 TENSORBOARD_NAME=projects/project-id/locations/us-central1/tensorboards/910xxxxx16
 WID_KSA=wid-sa
 ARGS=steps=150 per_device_batch_size=6 enable_checkpointing=false enable_profiler=false remat_policy=full max_target_length=2048  log_period=50 global_parameter_scale=8

ai-infrastructure/tpu-training-on-gke/examples/xpk/multi-slice-8b.sh.tmpl

@@ -3,4 +3,4 @@ set -e
 
 export LIBTPU_INIT_ARGS="--xla_tpu_enable_data_parallel_all_reduce_opt=true --xla_tpu_data_parallel_opt_different_sized_ops=true --xla_tpu_enable_async_collective_fusion=true --xla_tpu_enable_async_collective_fusion_fuse_all_gather=true --xla_tpu_enable_async_collective_fusion_multiple_steps=true --xla_tpu_overlap_compute_collective_tc=true --xla_enable_async_all_gather=true"
 
-python3 MaxText/train.py MaxText/configs/base.yml run_name=maxtext-single-slice-201 dataset_path=gs://prefix-artifact-repository/datasets base_output_directory=gs://prefix-artifact-repository/runs steps=150 log_period=50 per_device_batch_size=6 global_parameter_scale=8 enable_checkpointing=false enable_profiler=false remat_policy=full dcn_data_parallelism=2 ici_fsdp_parallelism=16 
+python3 MaxText/train.py MaxText/configs/base.yml run_name=maxtext-single-slice-201 dataset_path=gs://your-prefix-artifact-repository/datasets base_output_directory=gs://your-prefix-artifact-repository/runs steps=150 log_period=50 per_device_batch_size=6 global_parameter_scale=8 enable_checkpointing=false enable_profiler=false remat_policy=full dcn_data_parallelism=2 ici_fsdp_parallelism=16

ai-infrastructure/tpu-training-on-gke/examples/xpk/single-slice-8b.sh.tmpl

@@ -3,4 +3,4 @@ set -e
 
 export LIBTPU_INIT_ARGS="--xla_tpu_enable_data_parallel_all_reduce_opt=true --xla_tpu_data_parallel_opt_different_sized_ops=true --xla_tpu_enable_async_collective_fusion=true --xla_tpu_enable_async_collective_fusion_fuse_all_gather=true --xla_tpu_enable_async_collective_fusion_multiple_steps=true --xla_tpu_overlap_compute_collective_tc=true --xla_enable_async_all_gather=true"
 
-python3 MaxText/train.py MaxText/configs/base.yml run_name=maxtext-single-slice-201 dataset_path=gs://prefix-artifact-repository/datasets base_output_directory=gs://prefix-artifact-repository/runs steps=150 log_period=50 per_device_batch_size=6 global_parameter_scale=8 enable_checkpointing=false enable_profiler=false remat_policy=full dcn_data_parallelism=1 ici_fsdp_parallelism=16 
+python3 MaxText/train.py MaxText/configs/base.yml run_name=maxtext-single-slice-201 dataset_path=gs://your-prefix-artifact-repository/datasets base_output_directory=gs://your-prefix-artifact-repository/runs steps=150 log_period=50 per_device_batch_size=6 global_parameter_scale=8 enable_checkpointing=false enable_profiler=false remat_policy=full dcn_data_parallelism=1 ici_fsdp_parallelism=16