What problem does it solve?
This Skill addresses the challenge of efficiently processing and transforming large datasets for machine learning workloads, especially when datasets exceed single-machine memory or require distributed computation.
Core Features & Use Cases
- Distributed Data Processing: Scales data ingestion, transformation, and preprocessing across multiple CPU or GPU nodes.
- Flexible Data Formats: Supports reading and writing various formats including Parquet, CSV, JSON, and images.
- ML Framework Integration: Seamlessly integrates with Ray Train, PyTorch, and TensorFlow for end-to-end ML pipelines.
- Use Case: Use this Skill to preprocess terabytes of image data for a computer vision model, distributing the image augmentation and resizing tasks across a cluster of machines with GPUs.
Quick Start
Use the ray-data skill to read parquet files from 's3://my-bucket/data/' and apply a lowercasing transformation to the 'text' column.