NeMo Curator(文本)
NVIDIA 基于 RAPIDS 的 GPU 加速数据策展与去重工具
- 成立
- 2023-08
- 总部
- NVIDIA
- 数据类型
- 网页文本;预训练语料
- 规模口径
- 公司口径:RedPajama v2 子集模糊去重 3×H100 节点 0.65 小时,CPU 方案 10.7 小时
- License
- Apache-2.0
- 链接
- github.com
原理
NeMo Curator 把精确、模糊与子串去重放到 GPU(RAPIDS 加 Ray)上执行,同时提供可配置的启发式过滤、分类器集成与基于 LLM 的合成改写 [1]。
代表实现与口径
2023-08 的 NVIDIA 博客称 GPU 去重比 CPU 版快 20 倍、便宜 5 倍:4.5TB RedPajama 去重,CPU 集群 37 小时,4 个 DGX A100 节点 3 小时 [2]。当前 README 的口径不同:RedPajama v2 子集上 3×H100 节点 0.65 小时对 CPU 10.7 小时(约 16 倍),总拥有成本 CPU 方案 $315、NeMo Curator $190(约低 40%)[1]。均为公司口径,两组数据的硬件、数据集与基线都不同,不可互相换算;第三方站点流传的另一套倍数无一手出处,不采用。
谁在用
Nemotron-CC / v2(NVIDIA)(6.3T token,其中约 1.9T 为改写产物)的构建用到了分类器集成加合成改写的思路 [3]。
局限
依赖 GPU 集群;各工具吞吐数字不可跨家直接比较。
来源
相关卡片
卡片整理 2026-10-15。公司自报数据标“公司口径”;⚠️ 为未经独立核实。