TL;DR — Ray is the distributed-compute engine behind a lot of modern AI (training, tuning, batch + online inference). KubeRay is the Kubernetes operator that runs it: three CRDs — RayCluster, RayJob, RayService — turn "stand up a Ray cluster, run my job, serve my model" into declarative YAML, with autoscaling, fault tolerance, and zero-downtime upgrades handled for you.
What it is
Ray is an open-source framework for distributed Python — it spreads tasks and actors across a cluster, and ships libraries for training, tuning (Ray Tune), and serving (Ray Serve). KubeRay is the official operator that makes Ray a first-class Kubernetes citizen: you describe what you want as custom resources and the operator reconciles the Ray cluster for you. In the AI Native landscape it's in AI Native Infra › Orchestration and Scheduling.
Why it exists
Running Ray by hand on Kubernetes means wiring head/worker pods, services, autoscaling, and failure recovery yourself — and redoing it every time. KubeRay encodes all that operational knowledge into an operator, so a Ray cluster becomes as easy to declare as a Deployment, and integrates with the rest of the K8s ecosystem (schedulers, monitoring, ingress).
The three CRDs
| CRD | What it manages |
|---|---|
RayCluster | The full lifecycle of a Ray cluster — head + worker groups, autoscaling, fault tolerance. The base everything else builds on. |
RayJob | Creates a RayCluster, submits a job when it's ready, and (optionally) tears the cluster down when the job finishes. Ideal for training/batch runs. |
RayService | A RayCluster + a Ray Serve deployment graph, with high availability and zero-downtime upgrades. For online/LLM inference. |
Fig 1 — One operator reconciles Ray clusters for jobs, compute, and serving.
Ecosystem integration
KubeRay slots into the rest of the stack rather than reinventing it: pair it with a batch scheduler (Volcano, YuniKorn) or queueing (Kueue) for gang scheduling and quota; export metrics to Prometheus/Grafana and profile with py-spy; front RayService with an ingress or gateway. Since v1.3 the kubectl ray plugin smooths common workflows.
Quick start
Install the operator with Helm, then apply a RayCluster (or RayJob/RayService):
helm repo add kuberay https://ray-project.github.io/kuberay-helm/
helm install kuberay-operator kuberay/kuberay-operator -n ray-system --create-namespace
kubectl ray create cluster my-cluster # kubectl ray plugin (v1.3+)
kubectl get rayclustersFor a one-shot training run you'd apply a RayJob with shutdownAfterJobFinishes: true so the cluster is created, used, and reclaimed automatically.
When to use, when to skip
Use it when your AI workload is already (or should be) a Ray app — distributed training, hyperparameter sweeps, large batch inference, or Ray Serve for online/LLM inference — and you run on Kubernetes. RayService's zero-downtime upgrades make it a strong serving base.
Skip it if you're not using Ray — a plain Deployment or a dedicated serving runtime like KServe may be simpler. And remember KubeRay manages Ray clusters; it leans on Volcano/Kueue for cluster-wide gang scheduling and quota.
vs the alternatives
| Tool | Best for | Trade-off |
|---|---|---|
| KubeRay | Running Ray apps (train/tune/serve) on K8s | Only if you're on Ray |
| KServe | Standardized model serving/inference | Not general distributed compute |
| Kubeflow | Broader ML platform / pipelines | Heavier, more components |
| Volcano / Kueue | Scheduling/quota under KubeRay | Complementary, not replacements |
References
- Ray on Kubernetes (KubeRay) — official docs.
- Getting started — install + first cluster.
- RayJob quickstart — run a job end to end.
- ray-project/kuberay — source + examples.
Extra reads
- Ray on GKE overview — managed perspective.
- KubeRay on ACK — distributed AI architecture.
- Ray Serve — the serving layer behind RayService.
Verified against the official Ray/KubeRay docs (docs.ray.io), May 2026.
Where KubeRay fits: the mental model
KubeRay is a compute or serving layer that places expensive AI workloads and turns models into reliable runtime services. The useful question is not simply “can it run the demo?” It is whether the component gives your team a clear ownership boundary, predictable failure behavior, and enough evidence to operate changes safely. Treat it as one replaceable layer in a larger system rather than letting it quietly become the architecture.
Start by drawing the request and data path. Mark where untrusted input enters, where identity is checked, where durable state changes, and where retries can repeat work. That diagram tells you which guarantees belong to KubeRay and which still belong to your application, platform, cloud provider, or database. The distinction matters during incidents: a healthy process is not proof that the end-to-end task is correct.
Core concepts you should understand first
The vocabulary below is more important than any single SDK method. It lets application engineers, platform engineers, security reviewers, and incident responders describe the same system without confusing a framework feature with an end-to-end guarantee.
| Concept | Meaning in this layer | Design question |
|---|---|---|
| Resource request | Capacity reserved for placement; inaccurate requests create pending work or stranded accelerators. | Write down how KubeRay represents or enforces this before production. |
| Topology | NUMA, PCIe, NVLink, rack, and zone relationships that can dominate distributed workload performance. | Write down how KubeRay represents or enforces this before production. |
| Batching | Combining requests or examples to improve accelerator utilization at the cost of queueing latency. | Write down how KubeRay represents or enforces this before production. |
| Parallelism | Splitting model weights, pipeline stages, data, or requests across devices and processes. | Write down how KubeRay represents or enforces this before production. |
| Preemption | Reclaiming resources from lower-priority work; safe jobs need checkpoint and resume semantics. | Write down how KubeRay represents or enforces this before production. |
| Cold start | Time to schedule, pull images, load weights, compile kernels, and become ready. | Write down how KubeRay represents or enforces this before production. |
From quick start to a production deployment
The earlier quick start proves that the package or service runs. Production readiness is a different exercise. Build the smallest vertical slice that crosses every real boundary—identity, network, persistence, upstream provider, telemetry, and rollback—before broadening the feature set.
- Pin the compatibility envelope. Record the KubeRay release, language/runtime version, client SDK version, model or backend version, and—where applicable—Kubernetes API or driver requirements. Use a lock file, immutable image digest, or chart version; floating “latest” tags prevent repeatable rollback.
- Define contracts before configuration. Write the accepted input, successful output, error classes, timeout, idempotency behavior, and ownership of durable state. Validate at the boundary so corrupt work fails early instead of surfacing deep in a workflow.
- Create separate development, staging, and production identities. Do not copy a broad personal API key into every environment. Prefer workload identity or short-lived credentials, scope access by tenant and operation, and verify denial cases as part of deployment.
- Add bounded failure behavior. Every remote call needs a deadline. Retry only transient, idempotent operations with exponential backoff and jitter. Set concurrency and queue limits so an upstream slowdown becomes controlled backpressure rather than resource exhaustion.
- Instrument the complete path. Emit a correlation ID, component and release version, duration, outcome, retry count, and resource or cost dimensions. Keep sensitive prompt, document, and credential values out of ordinary logs.
- Ship through a reversible rollout. Run compatibility and regression tests, deploy to a canary or isolated workload, compare service-level indicators, then increase exposure. Preserve the previous artifact and configuration until rollback has been exercised.
Production configuration checklist
- Pin artifacts by version and, where possible, digest.
- Set connect, request, and total workflow deadlines.
- Bound retries, concurrency, queue length, and payload size.
- Separate read-only operations from mutations.
- Use idempotency keys for replayable mutations.
- Persist canonical state outside disposable workers.
- Encrypt traffic and durable data with managed keys.
- Redact secrets, tokens, prompts, and personal data.
- Apply per-tenant quotas and authorization filters.
- Expose readiness separately from process liveness.
- Back up metadata and test restore, not only backup.
- Document owner, escalation path, RPO, and RTO.
Failure modes and the response you should design
| Failure mode | What you observe | Engineering response |
|---|---|---|
| Unschedulable gang | Some workers start but the full distributed job cannot fit. | Use gang scheduling or admission so the group starts together. |
| Topology penalty | Workers span slow links or cross zones. | Express topology constraints and measure collective communication. |
| Memory fragmentation | Free memory exists but a large allocation fails. | Tune allocation/batching and recycle workers under controlled policy. |
| Driver mismatch | Host driver, runtime, CUDA, and framework are incompatible. | Qualify an immutable compatibility matrix before rollout. |
| Cold-start spike | Scale-out misses the latency objective. | Pre-pull images, cache weights, keep warm capacity, and measure each phase. |
| Noisy neighbor | One workload consumes shared network, CPU, or storage. | Apply quotas, priorities, isolation, and per-tenant saturation metrics. |
Turn these rows into runbook entries with an alert, first diagnostic query, safe mitigation, and escalation owner. Test at least one failure in staging every release cycle. If the system cannot be forced into a failure safely, it is usually not yet observable or isolated enough.
Security, privacy, and tenant isolation
Place KubeRay in a threat model, not just an architecture diagram. Identify human users, workload identities, administrators, upstream services, model providers, artifact registries, and data stores. For each edge, document authentication, authorization, encryption, audit evidence, and the consequence of credential compromise.
Apply least privilege at the operation and resource level. A component that only retrieves documents should not be able to delete the index; an evaluation worker should not inherit production mutation credentials; a model-serving pod should not need cluster-admin. In multi-tenant systems, enforce the tenant boundary before retrieval or execution and include tenant identity in quotas and audit events. Never rely on a prompt instruction, namespace string supplied by the client, or UI filtering as authorization.
Decide what data is permitted in telemetry. Prompts, retrieved chunks, tool arguments, model responses, notebooks, and traces can contain secrets or regulated data. Redact close to collection, keep high-sensitivity payload capture opt-in, encrypt exports, restrict support access, and give each class an explicit retention period. Verify deletion across caches, replicas, indexes, backups, and derived evaluation datasets.
Observability and service-level objectives
A useful dashboard follows the user-visible unit of work and then decomposes it by component, release, tenant tier, backend, and failure class. Start with these signals for KubeRay:
- accelerator utilization and memory — graph both rate and distribution, then compare with the previous release and traffic mix.
- queue wait and pending duration — graph both rate and distribution, then compare with the previous release and traffic mix.
- time to first token or first result — graph both rate and distribution, then compare with the previous release and traffic mix.
- throughput per device — graph both rate and distribution, then compare with the previous release and traffic mix.
- cold-start and model-load time — graph both rate and distribution, then compare with the previous release and traffic mix.
- failure, eviction, and preemption rate — graph both rate and distribution, then compare with the previous release and traffic mix.
Choose an SLO at the boundary your users experience, such as “99% of accepted tasks complete correctly within five minutes over 28 days.” Availability alone is insufficient for AI systems because a fast but incorrect or ungrounded result is still a failure. Pair latency and completion objectives with a reviewed quality or policy indicator. Page on rapid error-budget burn; use tickets for slow capacity trends.
Testing and release strategy
Use four layers. Unit tests cover deterministic adapters, schemas, policy, and error mapping without a live external service. Contract tests exercise the pinned integration boundary—API, CLI, SDK, protocol, or ephemeral service—and verify its exact surface. Scenario tests exercise representative end-to-end cases, including permissions and state. Load and resilience tests establish saturation, queue behavior, retry amplification, and recovery after dependency loss.
Keep a small blocking suite for every commit and a broader scheduled suite for expensive or probabilistic checks. Store results with the application version, KubeRay version, configuration hash, model/backend version, dataset version, and random seed. A score without that provenance cannot explain a regression. Before upgrading, read the migration notes, run both versions against the same replay set, and explicitly test rollback across any schema or state transition.
How to decide whether KubeRay is the right tool
| Question | Evidence to collect | Red flag |
|---|---|---|
| Does it remove a real constraint? | A measured bottleneck, missing guarantee, or repeated custom component. | Adoption is based only on a demo or feature count. |
| Can the team operate it? | Named owner, upgrade path, alerts, runbooks, backup, restore, and on-call skills. | Only the original prototype author understands failure behavior. |
| Is the interface portable? | Your domain contracts wrap vendor-specific APIs; data and state have an export path. | Business objects are inseparable from framework internals. |
| Does it meet the envelope? | Benchmarks using your payloads, concurrency, topology, quality bar, and cost model. | Published benchmark hardware or workload does not resemble production. |
| Is failure affordable? | Tested degraded mode, bounded blast radius, rollback, RPO, and RTO. | A component outage blocks unrelated tenants or irreversible actions. |
Prefer the smallest component that satisfies the required guarantees. A provider SDK, relational table, background job, or standard Kubernetes controller is often better than another platform when the workload is small and predictable. Choose KubeRay when its specific abstraction removes sustained engineering work and the team is willing to own its lifecycle.
A focused 90-minute validation lab
- Minutes 0–15: run the documented quick start in a disposable environment with pinned dependencies. Save the exact commands and a known-good input/output fixture.
- Minutes 15–35: replace the toy input with one representative case from your system. Add schema validation, a deadline, and a correlation ID.
- Minutes 35–55: force invalid credentials, a timeout, malformed input, and one dependency failure. Record the observed errors and whether retries are safe.
- Minutes 55–75: run a small concurrency test and capture latency, throughput, saturation, and unit cost. Do not extrapolate beyond the tested range.
- Minutes 75–90: write the adoption decision: required guarantees met, open risks, owner, next experiment, and the simplest credible alternative.
Frequently asked questions
Should we standardize on KubeRay for every team?
Standardize the contracts, telemetry, security controls, and release evidence first. Standardizing one implementation is useful only when workloads share requirements and a platform team owns upgrades and support.
Can we use the hosted version and skip operations work?
Hosted service removes part of the control-plane burden, not architecture ownership. You still own identity, tenant isolation, data classification, quotas, dependency failure, observability, export, and an exit plan.
What should be pinned for reproducibility?
Pin the tool/server, client SDK, runtime, configuration, model or backend, container image digest, and test dataset. Record these values with every benchmark and evaluation result.
When is a proof of concept ready for production?
After representative success and failure tests pass, sensitive data paths are approved, limits and SLOs are defined, telemetry and runbooks exist, restore or rollback is rehearsed, and an accountable owner accepts the remaining risk.
Official sources and freshness
This guide was reviewed for architecture and operational guidance on 10 July 2026. Projects evolve quickly: verify installation syntax, supported versions, feature maturity, and upgrade notes against the exact release you deploy.