NeMo Curator
GPU accelerated toolkit for filtering, deduplicating, and transforming multimodal AI training data.
About
NVIDIA's answer to data curation bottlenecks moves the whole pipeline onto GPUs: NeMo Curator uses RAPIDS libraries such as cuDF, cuML, and cuGraph, orchestrated over Ray based executors, to filter, classify, deduplicate, and transform training data across text, image, video, and audio modalities. The team reports fuzzy deduplication of a RedPajama v2 subset dropping from 10.7 hours to 0.65 hours, roughly a 16x speedup, on three H100 nodes compared with CPU alternatives. Text tooling covers language identification, quality classifiers, and exact, fuzzy, and semantic dedup; images get aesthetic and NSFW filtering plus embedding generation; video supports scene detection and clip extraction; audio adds transcription and quality filtering. The same stack powers the data pipelines behind NVIDIA's Nemotron model family. The GPU path expects Linux x86_64, CUDA 12, and about 16 GB of GPU memory, with installation through uv packages or Docker containers. The code is Apache 2.0 licensed with about 1,700 GitHub stars, aimed squarely at teams curating pretraining scale corpora on GPU clusters.
Should you use NeMo Curator?
Pick it when
Pick it when you curate pretraining-scale corpora on NVIDIA GPU clusters and CPU-based dedup or classification has become the bottleneck, especially if you also process video or audio and already run Linux with CUDA 12.
Look elsewhere when
Its GPU path expects Linux x86_64, CUDA 12, and about 16 GB of GPU memory; on Macs, Windows, or CPU-only clusters the speedup that justifies it is gone, and CPU-first DataTrove or Data-Juicer are easier at difficulty 3 versus 4.
Alternatives to NeMo Curator
- DataTrove
Runs on CPU locally or on Slurm and ships the FineWeb pipeline; text only and without GPU speedups, but no CUDA or VRAM requirement.
- Data-Juicer
CPU first with optional GPU for model-based operators and 200+ multimodal operators on Ray; an easier hardware fit, minus RAPIDS acceleration.
Reviews (0)
Leave a Review
No reviews yet. Be the first to review!
Details
- Category
- Datasets & Training
- Price
- Free
- Platform
- Local/Desktop
- Difficulty
- Advanced (4/5)
- License
- Apache-2.0
- Minimum VRAM
- 16 GB
- Added
- Jul 29, 2026