What problem does it solve?
Ray Data simplifies processing, transforming, and streaming very large machine learning datasets that exceed single-machine memory, removing the friction of distributed preprocessing and batch inference so teams can move from local experiments to cluster-scale production pipelines.
Core Features & Use Cases
- Streaming execution for datasets larger than memory, enabling iterative processing with bounded resources.
- GPU-accelerated transforms to speed up image/audio preprocessing and other heavy compute steps.
- Multi-format ingestion and output including Parquet, CSV, JSON, images, NumPy, and Pandas interoperability for ETL and model training workflows.
- Framework integration with Ray Train, PyTorch, and TensorFlow for seamless data->training pipelines and distributed batch inference.
- Use case: Run distributed preprocessing and feature engineering over hundreds of gigabytes stored on S3, then feed the processed Parquet shards into a Ray Train job for scalable model training.
Quick Start
Load a Parquet dataset from s3://bucket/data/*.parquet and run a batched lowercase text transform with a batch size tuned for your cluster.