From 3e249e6e3354dd53627baf7ed665bf31e04bf866 Mon Sep 17 00:00:00 2001 From: Ryan O'Leary Date: Tue, 21 Nov 2023 18:53:11 +0000 Subject: [PATCH] Added link to webhook README in TPU_GUIDE --- ray-on-gke/TPU_guide.md | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/ray-on-gke/TPU_guide.md b/ray-on-gke/TPU_guide.md index 014697594..25c48cedc 100644 --- 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).