Live
Smartsheet MCP integration: connecting Amazon Quick and Kiro on AWSLocal AI Hardware Expands with RTX Spark, DGX Spark Clusters, and Upcoming Windows DGX StationAI SRE context halves diagnosis latency and reduces tool callsGitHub CLI adds optional expiring OAuth tokens with automatic refreshDocker Sandbox + BAND AI Kit Enables Collaborative Agents in Isolated MicroVMsObjectRef GA: SQL‑Native Access to Cloud Storage Files for Multimodal PipelinesPractical Guide to IAM Roles Anywhere PKI Integration with Exportable ACM CertificatesReplacing Round‑Robin Load Balancing with GKE Inference Gateway to Quadruple Batch Inference ThroughputSmartsheet MCP integration: connecting Amazon Quick and Kiro on AWSLocal AI Hardware Expands with RTX Spark, DGX Spark Clusters, and Upcoming Windows DGX StationAI SRE context halves diagnosis latency and reduces tool callsGitHub CLI adds optional expiring OAuth tokens with automatic refreshDocker Sandbox + BAND AI Kit Enables Collaborative Agents in Isolated MicroVMsObjectRef GA: SQL‑Native Access to Cloud Storage Files for Multimodal PipelinesPractical Guide to IAM Roles Anywhere PKI Integration with Exportable ACM CertificatesReplacing Round‑Robin Load Balancing with GKE Inference Gateway to Quadruple Batch Inference Throughput
Google Cloud

Distributed LLM Inference on GKE: llm-d 1.0 Brings Cache‑Aware Routing and Multi‑Accelerator Pools

AI SummaryPowered by AI

Google and partners released llm-d 1.0, a Kubernetes‑native stack that coordinates LLM inference across GPUs and TPUs with cache‑aware routing and tiered KV caches. The change lets engineers serve multi‑step agents at scale without duplicate work, improving latency, utilization, and upgrade safety.

Google’s open‑source llm-d 1.0 rewrites how large language model inference is delivered on Kubernetes by turning a collection of model servers into a single, cache‑aware serving pool that can span both GPUs and TPUs. For engineers building agent‑centric workloads, the shift eliminates redundant prompt recomputation, reduces accelerator memory pressure, and provides a unified endpoint that adapts to bursty, multi‑turn traffic.

Why Agent‑Centric Workloads Require a New Approach

Traditional inference pipelines assumed a one‑prompt‑one‑response pattern. Modern agents generate a series of calls, often interleaving interactive turns with background reinforcement‑learning rollouts. When such traffic is load‑balanced round‑robin, each replica repeats the same prefix computation, fills its accelerator memory with duplicate state, and can become a bottleneck during bursty loops. The result is higher latency for interactive users and wasted compute.

Key Architectural Changes in llm-d 1.0

llm-d introduces three core mechanisms that address the above problems:

  • Cache‑aware routing: The router tracks which replica holds a given KV cache prefix and directs subsequent steps of the same session to that replica, avoiding recompute.
  • Tiered KV cache: Cache entries can be offloaded from accelerator memory, extending the effective context window without exhausting device RAM.
  • Priority flow control: Bursty agent loops are queued behind latency‑sensitive interactive turns, preserving response times for end users.

These mechanisms sit on top of existing PyTorch serving engines (vLLM, SGLang) and are exposed through a single API endpoint. The stack also supports prefill/decode disaggregation, wide expert parallelism, and batch inference across mixed accelerator fleets.

Operational Implications

From an operations perspective, llm-d 1.0 offers several practical benefits:

  • Multi‑tenant QoS: Priority‑aware routing and flow control let different workloads share the same pool while meeting their own service‑level objectives.
  • Stable API surface: Version‑1 APIs and a hardened request path simplify integration across clouds and accelerator types.
  • Zero‑downtime upgrades: The router remains highly available, and the DisaggregatedSet component coordinates rolling updates that respect in‑flight prefill/decode state.
  • Composable deployment: Teams can keep their existing model servers and clusters, adding only the llm-d routing and cache components they need.

Because the router abstracts the underlying accelerator pool, adding capacity, mixing GPU generations, or moving a model to a TPU fleet is a platform change rather than a code change in the agents themselves.

Security and Reliability Considerations

Consolidating traffic behind a single endpoint reduces the number of exposed services, but it also concentrates routing logic. Practitioners should monitor router health, ensure high‑availability configurations, and validate that the KV cache index does not become a single point of failure. The zero‑downtime update path mitigates disruption risk, yet any misconfiguration in DisaggregatedSet could lead to context loss for active sessions, so thorough testing in a staging environment is advisable.

Related CloudNinjas coverage: Google Cloud.

What This Means For Practitioners

Adopting llm-d 1.0 means re‑architecting inference pipelines to rely on a coordinated router rather than independent load‑balanced pods. Engineers should evaluate existing model server compatibility with vLLM or SGLang, plan for KV cache sizing on accelerator memory, and configure flow‑control policies that reflect their latency versus throughput priorities. Operationally, set up health checks for the router, test rolling updates with DisaggregatedSet, and treat the cache index as a critical component that requires monitoring and backup strategies.

Originally published atGoogle Cloud Blog