Key Features#

NeMo Curator is an enterprise-grade platform for scalable, privacy-aware data curation across text, image, video, and audio. It empowers teams to prepare high-quality, compliant datasets for LLM and AI training, with robust support for distributed, cloud-native, and on-premises workflows. Leading organizations trust NeMo Curator for its modular pipelines, advanced filtering, and seamless integration with modern MLOps environments.

Why NeMo Curator?#

  • Trusted by leading organizations for LLM and generative AI data curation

  • Open source, NVIDIA-supported, and actively maintained

  • Seamless integration with enterprise MLOps and data platforms

  • Proven at scale: from laptops to multi-node GPU clusters

Benchmarks & Results#

  • Deduplicated 1.96 trillion tokens in 0.5 hours using 32 NVIDIA H100 GPUs (RedPajama V2 scale)

  • Up to 80% data reduction and significant improvements in downstream model performance (see ablation studies)

  • Efficient curation of Common Crawl: from 2.8TB raw to 0.52TB high-quality data in under 38 hours on 30 CPU nodes


Text Data Curation#

NeMo Curator offers advanced tools for text data loading, cleaning, filtering, deduplication, and classification. Built-in modules support language identification, quality estimation, domain and safety classification. Pipelines are fully modular and can be customized for diverse NLP and LLM training needs.

Data Loading

Efficiently load and manage massive text datasets, with support for common formats and scalable streaming.

Data Loading Concepts
Data Processing

Advanced filtering, deduplication, classification, and pipeline design for high-quality text curation.

Text Processing Concepts
Text Curation Quickstart

Set up your environment and run your first text curation pipeline with NeMo Curator.

Get Started with Text Curation

Image Data Curation#

NeMo Curator supports scalable image dataset loading, embedding, classification (aesthetic, NSFW, etc.), filtering, deduplication, and export. It leverages state-of-the-art vision models (for example, CLIP, timm) with pipeline-based architecture for efficient GPU-accelerated processing. Modular pipelines enable rapid experimentation and integration with text and multimodal workflows.

Data Loading

Load and manage large-scale image datasets for curation workflows.

Data Loading Concepts (Image)
Data Processing

Embedding generation, classification (aesthetic, NSFW), filtering, and deduplication for images.

Data Processing Concepts (Image)
Data Export

Export, save, and reshard curated image datasets for downstream use.

Data Export Concepts (Image)
Image Curation Quickstart

Set up your environment and install NeMo Curator’s image modules.

Get Started with Image Curation

Audio Data Curation#

NeMo Curator provides speech and audio curation capabilities designed for preparing high-quality speech datasets for ASR model training and multimodal applications. It leverages pretrained .nemo model checkpoints via the NeMo Framework for transcription, quality assessment through Word Error Rate (WER) calculation, and seamless integration with text curation workflows.

Data Loading

Load and manage audio datasets with manifests, file paths, and transcriptions for curation workflows.

Dataset Manifests and Ingest
ASR Processing

Automatic speech recognition inference, quality assessment, and transcription using NeMo Framework models.

ASR Pipeline Architecture
Quality Assessment

Word Error Rate (WER) calculation, duration analysis, and quality-based filtering for speech data.

Audio Quality Metrics
Audio Curation Quickstart

Set up your environment and run your first audio curation pipeline with NeMo Curator.

Get Started with Audio Curation

Video Data Curation#

NeMo Curator provides distributed video curation pipelines, supporting scalable data flow, pipeline stages, and efficient processing for large video corpora. The architecture supports high-throughput, cloud-native, and on-prem deployments.

Architecture

Distributed processing, Ray-based foundation, and autoscaling for video curation.

Architecture
Key Abstractions

Stages, pipelines, and execution modes in video curation workflows.

Key Abstractions
Data Flow

How data moves through the system, from ingestion to output, for efficient large-scale video curation.

Data Flow
Video Curation Quickstart

Set up your environment and run your first video curation pipeline with NeMo Curator.

Get Started with Video Curation

Deployment and Integration#

NeMo Curator is designed for distributed, cloud-native, and on-premises deployments. It integrates easily with your existing MLOps pipelines. Modular APIs enable flexible orchestration and automation.

Deployment Options

See the Admin Guide for deployment guidance and infrastructure recommendations.

About Setup & Deployment
Memory Management

Optimize memory usage and partitioning for large-scale curation workflows.

Memory Management Guide
GPU Acceleration

Leverage NVIDIA GPUs for faster data processing and pipeline acceleration.

GPU Processing Guide
Resumable Processing

Continue interrupted operations and recover large dataset processing with checkpointing and batching.

Resumable Processing