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.
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
Related Tools
Fine-tune LLMs 2x faster with 80% less memory
Tool for fine-tuning LLMs with various configurations
Open source data labeling platform for ML projects
Library for accessing and sharing ML datasets
Hugging Face library for parameter-efficient fine-tuning
Python library for synthetic data generation and curation pipelines with batch LLM inference.