[ ABORT TO HUD ]
SEQ. 1
SEQ. 2
Google Kubernetes Engine for AI
🚢 GKE & TPUs for AI 12m 250 BASE XP⌨ HANDS-ON LAB
Orchestrating High-Performance AI Infrastructure
While Vertex AI manages serverless endpoints, large-scale enterprise training and ultra-low-latency custom serving (such as vLLM, TensorRT-LLM, or distributed Ray clusters) often require bare-metal orchestration on Google Kubernetes Engine (GKE).
Key GKE AI Platform Innovations
- Ray on GKE (KubeRay): Deploys and manages elastic Ray clusters for distributed model training, batch embeddings, and reinforcement learning.
- Kueue: Kubernetes-native job queuing that fairly allocates GPU/TPU quota across engineering teams with priority preemption.
- Node Auto-Provisioning (NAP): Automatically provisions and tears down exact GPU/TPU machine types (e.g.,
a3-highgpu-8gwith 8x NVIDIA H100 GPUs) based on pending pod requests.
# vLLM High-Throughput Inference Deployment on GKE
apiVersion: apps/v1
kind: Deployment
metadata:
name: vllm-gemini-gemma-inference
spec:
replicas: 2
template:
spec:
containers:
- name: vllm
image: vllm/vllm-openai:latest
args: ["--model", "google/gemma-2-27b-it", "--tensor-parallel-size", "4", "--gpu-memory-utilization", "0.95"]
resources:
limits:
nvidia.com/gpu: "4"
volumeMounts:
- mountPath: /dev/shm
name: dshm
volumes:
- name: dshm
emptyDir:
medium: Memory
⌨ HANDS-ON LABInspect GPU/TPU Node Pools on GKE for Model Serving
⭐ +150 XPInspect Google Kubernetes Engine (GKE) accelerator node labels for hosting open-source LLM inference with vLLM.
1Filter GKE nodes by the cloud.google.com/gke-accelerator label.
OBJECTIVE 1 / 1 — type "hint" if stuck
SYNAPSE VERIFICATION
QUERY 1 // 2
What is the primary benefit of using GKE for AI workloads?
It is a no-code platform
It provides highly scalable, containerized orchestration for custom GPU/TPU infrastructure
It replaces the need for a database
It is only used for frontend web hosting