What problem does it solve?
This Skill addresses the need for fast, GPU-accelerated data curation, allowing users to efficiently prepare high-quality training datasets for large language models (LLMs).
Core Features & Use Cases
- Multi-modal Data Support: Process text, images, video, and audio with support for various file types.
- GPU Acceleration: Leverage GPU power for fuzzy deduplication, quality filtering, and more.
- Fuzzy Deduplication: Quickly deduplicate data up to 16x faster than CPU alternatives.
- Quality Filtering: Apply 30+ heuristic filters for comprehensive dataset refinement.
- Semantic Deduplication: Remove semantically similar documents for precise dataset creation.
- PII Redaction: Protect sensitive information by redacting personally identifiable information.
- NSFW Detection: Identify and filter out inappropriate content to ensure compliance.
- Use Case: Use NeMo Curator to prepare high-quality LLM training data from Common Crawl web scrapes, web data, or large corpora.
Quick Start
Use the nemo-curator skill to prepare training data from a web scrape using 'common_crawl/*' and save it as 'curated_data.parquet'.