Back Home

AI 基礎設施

SageMaker HyperPod Adds Managed Ray: Training, Inference, and Failure Recovery Share the KubeRay Interface

AWS has integrated Ray cluster creation, remote job submission, monitoring, and workspace connectivity into HyperPod on EKS. The interface retains KubeRay and standard Ray APIs, but deployment still requires several add-ons, and AWS has not published cost or performance comparisons.

Delince · CC BY-SA 3.0 · Image source
zh-Hant

AWS has added managed Ray capabilities to the EKS path for SageMaker HyperPod, bringing tasks that previously required users to write Kubernetes manifests, build images, configure port forwarding, and set up Prometheus/Grafana into SageMaker Studio. Users can create RayCluster resources through the interface, access Ray Dashboard and Amazon Managed Grafana, submit distributed jobs, and configure stalled job detection. Under the hood, the open-source KubeRay project continues to manage RayCluster, RayJob, and RayService resources, so existing Ray Train, Ray Serve, and standard submission API code should generally require no changes.

The development workflow also changes materially. JupyterLab or Code Editor spaces can connect to an existing cluster as workers without provisioned compute resources, with code connecting through `ray.init(address="auto")`. Dependencies can be injected through `runtime_env`, reducing the need to rebuild containers whenever packages change. Remote jobs use `toolkit-for-ray-on-sagemaker-ai`, which connects SageMaker cluster-name resolution and EKS IAM credentials to the Ray CLI. Dashboard URLs are short-lived, IAM-authenticated endpoints bound to their creator, making them better suited to multi-user environments than a publicly exposed Ray head service or long-running port forwarding.

For long-running AI workloads, HyperPod node health monitoring and automatic recovery work alongside Ray's fault-tolerance mechanisms. Training can use hierarchical checkpointing, first writing frequently updated state to the cluster's CPU memory and then periodically persisting it. Ray Serve can also load weights from JumpStart and offload the KV cache for long-context inference to tiered storage. However, this is not a one-click setup: prerequisites include EKS HyperPod, Spaces, Observability, KubeRay, and Ray Endpoint Operator. AWS has not provided benchmarks comparing startup time, failure recovery, or unit cost with self-managed Ray. Engineering teams should still measure the main-memory footprint of checkpoints, endpoint permission boundaries, and the effect of KV offloading on tail latency.

Sources

  1. Introducing new Ray capabilities on SageMaker HyperPod
  2. KubeRay: A toolkit to run Ray applications on Kubernetes