What problem does it solve?
This Skill addresses the challenge of efficiently processing and transforming large datasets for machine learning workloads, especially when data exceeds single-machine memory or requires distributed computation.
Core Features & Use Cases
- Scalable Data Processing: Handles datasets from gigabytes to terabytes across clusters.
- Streaming Execution: Processes data larger than available RAM.
- Multi-modal Support: Reads and processes various data types including Parquet, CSV, JSON, and images.
- Framework Integration: Seamlessly integrates with Ray Train, PyTorch, and TensorFlow for end-to-end ML pipelines.
- Use Case: Preprocess terabytes of image data for a computer vision model training job, distributing the transformations across a cluster of machines with GPU acceleration.
Quick Start
Use the ray-data skill to read all parquet files from 's3://my-data-bucket/raw/' and apply a lowercasing transformation to the 'text' column.