Added link to webhook README in TPU_GUIDE

GoogleCloudPlatform · Nov 21, 2023 · 3e249e6 · 3e249e6
1 parent e900253
commit 3e249e6
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/ray-on-gke/TPU_guide.md b/ray-on-gke/TPU_guide.md
@@ -106,6 +106,5 @@ def init_jax_from_ray(num_workers: int):
 init_jax_from_ray(num_workers=2)
 
 ``` 
-
-
+When initializing multi-host TPUs, the environment variables can be set using a mutating admission webhook. The webhook can be deployed following the instructions in the [README](https://github.com/GoogleCloudPlatform/ai-on-gke/tree/kuberay-tpu-env-injector/ray-on-gke/user/kuberay-tpu-webhook#readme).