Система обработки данных с более чем 200 комбинируемыми операторами для очистки, фильтрации, дедупликации и генерации текста, изображений, аудио и видео для обучения foundation-моделей.
NVIDIA-NeMo/Curator
Набор инструментов с GPU-ускорением для создания воспроизводимых конвейеров фильтрации, дедупликации и преобразования больших текстовых, графических, видео- и аудиодатасетов для обучения ИИ.
- Звёзды
- 1,818
- Форки
- 351
- Открытые issues
- 297
- Последний коммит
- 8 часов назад
Что это?
NeMo Curator — это Python-инструментарий для подготовки больших датасетов, используемых для обучения ИИ-моделей. ML-инженеры и команды по работе с данными могут собирать конвейеры из этапов загрузки, фильтрации, дедупликации, классификации и записи данных. Для текста, изображений, видео и аудио предусмотрены отдельные наборы компонентов.
Этапы конвейера объявляют собственные требования к CPU, GPU и числу реплик, а подключаемый исполнитель распределяет их по ресурсам. Одно и то же описание конвейера можно запускать на одной машине или в многонодном кластере Ray, в том числе в HPC-средах под управлением Slurm. В документации проекта Xenna указан как исполнитель по умолчанию для production, а бэкенды на базе Ray доступны в экспериментальном режиме.
Инструментарий входит в набор NVIDIA NeMo и используется в конвейерах обработки данных, описанных для датасетов Nemotron и Nemotron-CC. Установить его можно через uv или Docker; для обработки текста на GPU в Linux предусмотрен вариант с CUDA 12.
Ключевые возможности
- 01Модульные этапы загрузки, фильтрации, дедупликации, классификации и записи данных.
- 02Дедупликация и классификация с GPU-ускорением на базе RAPIDS.
- 03Запуск одного и того же конвейера на ноутбуке или в многонодном кластере Ray.
- 04Плагины для работы с текстом, изображениями, видео и аудио.
- 05Встроенный сервер инференса, совместимый с OpenAI, для сценариев генерации синтетических данных.
Сценарии использования
- Дедупликация и фильтрация веб-текстов по качеству перед предварительным обучением LLM.
- Фильтрация и создание эмбеддингов для больших коллекций изображений, используемых в обучающих датасетах.
- Подготовка видеоклипов с обнаружением сцен и фильтрацией по движению.
- Транскрибирование и фильтрация речевых данных для обучения аудиомоделей.
Быстрый старт
- 1
Установите менеджер пакетов uv (скрипт установки из README передаётся в sh через конвейер, поэтому здесь команда не приводится)
- 2
Создайте виртуальное окружение и установите дополнительные компоненты для обработки текста на CPU
uv venv && source .venv/bin/activate uv pip install "nemo-curator[text_cpu]" - 3
Проверьте установку, выведя версию
python -c "import nemo_curator; print(nemo_curator.__version__)"
Что важно знать
- Для обработки текста на GPU требуется CUDA 12 в Linux x86_64 и около 16 ГБ памяти GPU.
- Стандартная установка через pip не поддерживается для текстового дополнения CUDA 12; для него нужен специальный файл переопределения uv.
- Для аудио- и видеопроцессов требуется FFmpeg; лучше всего использовать для этого пользовательскую сборку контейнера.
- Начиная с версии 26.09 NVIDIA больше не публикует новые образы контейнеров NeMo Curator в NGC.
Темы
Частые вопросы
- Что такое Curator?
- Набор инструментов с GPU-ускорением для создания воспроизводимых конвейеров фильтрации, дедупликации и преобразования больших текстовых, графических, видео- и аудиодатасетов для обучения ИИ. NeMo Curator — это Python-инструментарий для подготовки больших датасетов, используемых для обучения ИИ-моделей. ML-инженеры и команды по работе с данными могут собирать конвейеры из этапов загрузки, фильтрации, дедупликации, классификации и записи данных. Для текста, изображений, видео и аудио предусмотрены отдельные наборы компонентов.
- Curator можно использовать бесплатно? Какая у него лицензия?
- Curator распространяется с открытым исходным кодом под лицензией Apache-2.0. Перед использованием в коммерческих продуктах проверьте условия лицензии.
- Curator всё ещё поддерживается?
- Активно поддерживается. Последний коммит в Curator — 21 окт. 2026 г..
- Для чего можно использовать Curator?
- Дедупликация и фильтрация веб-текстов по качеству перед предварительным обучением LLM. Фильтрация и создание эмбеддингов для больших коллекций изображений, используемых в обучающих датасетах. Подготовка видеоклипов с обнаружением сцен и фильтрацией по движению. Транскрибирование и фильтрация речевых данных для обучения аудиомоделей.
- Как установить Curator?
- Установите менеджер пакетов uv (скрипт установки из README передаётся в sh через конвейер, поэтому здесь команда не приводится) Создайте виртуальное окружение и установите дополнительные компоненты для обработки текста на CPU: uv venv && source .venv/bin/activate uv pip install "nemo-curator[text_cpu]" Проверьте установку, выведя версию: python -c "import nemo_curator; print(nemo_curator.__version__)"
- Где скачать Curator?
- Исходный код и инструкции по установке — на GitHub: https://github.com/NVIDIA-NeMo/Curator
Описание создано ИИ на основе README проекта. Статистика GitHub обновляется ежедневно.
Похожие проекты
Python-фреймворк для создания RAG-приложений, агентов и систем обработки документов, связывающих LLM с вашими данными.
Система на Python для запуска LLM-пайплайнов map-reduce и ETL над большими объёмами структурированных и неструктурированных данных по описанию на естественном языке.
Компилятор машинного обучения и движок развёртывания, позволяющие запускать большие языковые модели на GPU, в браузерах и на мобильных устройствах с OpenAI-совместимыми API.
Лёгкая библиотека Python с единым API в стиле OpenAI для разных LLM-провайдеров и Agents API с поддержкой инструментов и MCP.
MNN — лёгкий движок Alibaba для инференса и обучения моделей глубокого обучения на мобильных и встроенных устройствах, включая запуск LLM прямо на устройстве.