Skip to content

feat(ray-on-eks): batch LLM inference blueprint with Iceberg + tokenomics#1124

Merged
nabuskey merged 1 commit into
mainfrom
ray-inference
Jul 14, 2026
Merged

feat(ray-on-eks): batch LLM inference blueprint with Iceberg + tokenomics#1124
nabuskey merged 1 commit into
mainfrom
ray-inference

Conversation

@vara-bonthu

@vara-bonthu vara-bonthu commented Jul 14, 2026

Copy link
Copy Markdown
Collaborator

What does this PR do?

🛑 Please open an issue first to discuss any significant work and flesh out details/direction. When we triage the issues, we will add labels to the issue like "Enhancement", "Bug" which should indicate to you that this issue can be worked on and we are looking forward to your PR. We would hate for your time to be wasted.
Consult the CONTRIBUTING guide for submitting pull-requests.

A complete, runnable batch LLM inference blueprint on the ray-on-eks stack. DeepSeek-R1-8B on vLLM 0.22 / Ray 2.56 (KubeRay v1.6), triaging mock support tickets into structured output (sentiment, urgency, category, summary).

  • Ships an end-to-end example (data-stacks/ray-on-eks/examples/ray-batch-inference/) with a one-command deploy.sh: stage weights to S3, mirror the image to ECR, generate data, and run inference.
  • Implements both batch inference patterns against the same dataset so the tradeoff is measurable. a shared RayService endpoint (pattern 1) and the native ray.data.llm engine-in-job (pattern 2).
  • Lands tickets in an Apache Iceberg table (Glue catalog) and reads them incrementally via a per-pattern watermark, so re-runs only process new rows (at-least-once, idempotent output).
  • Adds a Grafana dashboard + PodMonitor (DCGM GPU cost/utilization + Ray Data pipeline throughput) wired into the stack's kube-prometheus, all open-source.
  • Enables enable_nvidia_gpu_operator, SOCI FastImagePull + RAID0 on the GPU node class, and weight streaming from S3 via the Run:ai Model Streamer.
  • Publishes a full website guide with measured cold-start anatomy, a tokenomics cost comparison, and architecture diagrams.

Motivation

  • Teams that need both real-time and batch inference usually build two separate stacks; this shows Ray collapsing them onto one platform on EKS.
  • Batch on ephemeral GPUs is the cost-efficient pattern (rent the GPU only while the job runs) but only works if cold start is fast. the blueprint solves that with same-region ECR + S3 weight streaming.

More

  • Yes, I have tested the PR using my local account setup (Provide any test evidence report under Additional Notes)
  • Mandatory for new blueprints. Yes, I have added a example to support my blueprint PR
  • Mandatory for new blueprints. Yes, I have updated the website/docs or website/blog section for this feature
  • Yes, I ran pre-commit run -a with this PR. Link for installing pre-commit locally

For Moderators

  • E2E Test successfully complete before merge?

Additional Notes

…mics

End-to-end batch inference example on the ray-on-eks stack: DeepSeek-R1-8B
on vLLM 0.22 / Ray 2.56 (KubeRay v1.6), two patterns (shared RayService
endpoint and native ray.data.llm engine-in-job) over the same dataset.

- examples/ray-batch-inference: staging + ECR mirror jobs, RayService,
  bounded ticket generator writing to an Iceberg table (Glue catalog),
  both batch RayJobs with incremental watermark reads, Ray metrics
  PodMonitor, deploy.sh/cleanup.sh helpers, README.
- Weights streamed from S3 via Run:ai Model Streamer; image mirrored to
  same-region ECR; SOCI FastImagePull + RAID0 on the GPU node class.
- NVIDIA GPU operator (DCGM) enabled; Grafana dashboard for GPU cost and
  Ray Data pipeline throughput wired into kube-prometheus-stack.
- website: full guide with measured cold-start, tokenomics comparison,
  live-Grafana walkthrough, and architecture diagrams.
@vara-bonthu vara-bonthu requested a review from nabuskey July 14, 2026 16:15
@nabuskey nabuskey merged commit a95def0 into main Jul 14, 2026
8 checks passed
@nabuskey nabuskey deleted the ray-inference branch July 14, 2026 18:32
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants