NeMo Curator

GPU accelerated toolkit for filtering, deduplicating, and transforming multimodal AI training data.

Open SourceSelf HostedOffline CapableGPU Required (16GB+ VRAM)
0.0 (0)

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

Price
Free
Platform
Local/Desktop
Difficulty
Advanced (4/5)
License
Apache-2.0
Minimum VRAM
16 GB
Added
Jul 29, 2026

Tags