ray-data

Process large datasets for machine learning with Ray Data.

1|Updated Feb 21, 2026
One-click install
npx skills add https://github.com/tianhao909/AI-Research-SKILLs-cn --skill ray-data-tianhao909
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: ray-data
Source: https://github.com/tianhao909/AI-Research-SKILLs-cn/tree/main/05-data-processing/ray-data
Command: npx skills add https://github.com/tianhao909/AI-Research-SKILLs-cn --skill ray-data-tianhao909

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill requires ray[data], pyarrow, pandas, and includes references (resource) components.

What problem does it solve?

This Skill addresses the challenge of processing large datasets for machine learning, enabling efficient data preparation and transformation at scale.

Core Features & Use Cases

  • Distributed Data Processing: Scales data loading and transformation across multiple nodes.
  • Streaming Execution: Handles datasets larger than available memory.
  • Framework Integration: Seamlessly works with PyTorch, TensorFlow, and Ray Train.
  • Use Case: Preprocess terabytes of image data for a computer vision model training pipeline, distributing the workload across a cluster for faster iteration.

Quick Start

Use the ray-data skill to read all parquet files from 's3://my-bucket/data/' and convert them to a PyTorch dataset.

Frequently Asked Questions about ray-data

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

FAQPage Schema
How do I process large datasets for machine learning when they exceed available memory?

Streaming execution enables distributed data processing for machine learning workloads that exceed available memory. It handles large datasets by streaming execution across CPU and GPU nodes, efficiently processing terabytes of data without loading everything into memory at once.

Can I use Ray Data to preprocess image data for PyTorch model training?

Yes, Ray Data integrates seamlessly with PyTorch and TensorFlow for ML pipelines. It supports multi-modal data loading, allowing you to preprocess image data for computer vision model training and convert the results directly into a PyTorch dataset.

What is the best way to run distributed ETL pipelines on Parquet and CSV files?

Distributed ETL pipelines can be run using this Skill to read and process various data formats including Parquet, CSV, and JSON. It scales data loading and transformation across multiple nodes, distributing the workload across a cluster for faster iteration.

Does Ray Data support batch inference and data preprocessing across CPU and GPU?

Yes, Ray Data supports streaming execution across both CPU and GPU for batch inference and data preprocessing. It is designed to scale data processing effortlessly across available compute resources for machine learning workloads.

How do I convert Parquet files from an S3 bucket into a distributed ML dataset?

You can read all Parquet files directly from an S3 bucket and convert them into a PyTorch dataset. This Skill enables scalable data loading and transformation, integrating with ML frameworks to prepare data for training pipelines.