flyte-kuberay

Provision ephemeral Ray clusters per Flyte task execution via KubeRay.

Updated Aug 27, 2026
One-click install
npx skills add https://github.com/tylertitsworth/skills --skill flyte-kuberay
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: flyte-kuberay
Source: https://github.com/tylertitsworth/skills/tree/main/flyte-kuberay
Command: npx skills add https://github.com/tylertitsworth/skills --skill flyte-kuberay

SYSTEM DOCUMENTATION & REQUIREMENTS

What problem does it solve?

Seamlessly run Ray workloads as Flyte tasks by provisioning ephemeral Ray clusters per task execution via the KubeRay operator. The integration enables Flyte users to configure RayJobConfig (head and worker resources, autoscaling, runtime_env) and run Ray Core, Ray Data, Ray Train, or Ray Serve within Flyte workflows while ensuring clusters are created, used, and torn down per task run.

Core Features & Use Cases

  • Ephemeral Ray clusters per task: Flyte launches a dedicated Ray cluster for each execution and tears it down after completion.
  • Rich Ray integration: Supports Ray Core, Ray Data, Ray Train, and Ray Serve within Flyte tasks.
  • Flexible configuration: Customize head and worker node settings, autoscaling, pod templates, and runtime environments via RayJobConfig.
  • Safe operational flow: Manages per-task lifecycle and resources in a Kubernetes environment.

Quick Start

Run a Flyte task that uses the flytekitplugins-ray integration to submit your Ray function to an ephemeral cluster.

Frequently Asked Questions about flyte-kuberay

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I run Ray distributed training workloads as Flyte tasks?

Run Ray distributed training as Flyte tasks by provisioning ephemeral Ray clusters per task execution. This integration provisions a dedicated cluster for each run and automatically tears it down upon completion.

What do I need to configure RayJobConfig for Kubernetes Ray clusters?

Configure RayJobConfig to customize head and worker node resources, autoscaling, pod templates, and runtime_env. This setup safely manages per-task lifecycle and resources within a Kubernetes environment.

Does Flyte support Ray Data, Ray Train, and Ray Serve within workflows?

Flyte supports Ray Core, Ray Data, Ray Train, and Ray Serve within workflows. The integration uses the flytekitplugins-ray package to submit functions to an ephemeral cluster.

Do I need the KubeRay operator to run Ray jobs in Flyte?

Yes, running Ray jobs in Flyte requires a KubeRay operator installed, the flytekitplugins-ray package in the task image, and Flyte's Ray backend plugin enabled.

Can I autoscale Ray worker nodes per task execution in Flyte?

Yes, you can autoscale Ray worker nodes per task execution in Flyte. RayJobConfig allows specifying autoscaling settings alongside head and worker node configurations.