AIKraft/Skills/nemotron: настройка

nemotron: настройка

Plan, configure, and chain repo-native Nemotron customization steps into single-step or multi-step pipelines: curation, translation, SFT/PEFT (AutoModel or Megatron-Bridge), pretraining/CPT, RL alignment (DPO/RLVR/GRPO/RLHF), BYOB/MCQ benchmarks, checkpoint conversion, ModelOpt optimization, env profiles, and evaluation of trained checkpoints or existing/hosted endpoints. Use when a request names

nvidia official Другое

Что делает навык

Планирует, настраивает и объединяет встроенные шаги репозитория Nemotron в одно- или многошаговые конвейеры (DAG) для кастомизации моделей. Решает задачи от курирования данных и SFT/PEFT до RL-выравнивания, оптимизации ModelOpt и оценки чекпоинтов или хостинговых эндпоинтов.

Когда применять

  • Планирование и создание многошаговых конвейеров обучения (например, перевод -> SFT -> RLHF).
  • Настройка параметров для конкретных шагов: SFT, PEFT (AutoModel/Megatron-Bridge), pretraining/CPT, DPO/RLVR/GRPO/RLHF.
  • Оценка (benchmark/eval) существующих или хостинговых моделей/эндпоинтов без обучения.
  • Конвертация чекпоинтов и оптимизация моделей с помощью ModelOpt.
  • Профилирование окружения и проверка совместимости артефактов между шагами.

Как работает

  1. Определить цель запроса и выбрать подходящий шаг или набор шагов, используя references/CATALOG.md и references/ARTIFACTS.md.
  2. Собрать необходимые входные данные (пути к данным, ID моделей, GPU, профили окружения); запросить их у пользователя, если они отсутствуют.
  3. Спланировать DAG конвейера, проверив цепочки артефактов и соответствие параметров (токенизатор, seq_length, batch size) через references/PATTERNS.md и HARDWARE.md.
  4. Создать новые YAML-конфиги в директории config выбранного шага, копируя схему из default.yaml, но не редактируя исходные файлы.
  5. Сформировать итоговую команду или набор команд, используя данные из references/COMMANDS.md и активного профиля окружения (NEMOTRON_ENV_FILE).
  6. Проверить безопасность (не раскрывать секреты) и валидность конфигурации перед выдачей результата.

Примеры запросов агенту

Собери конвейер: загрузи данные, выполни SFT через Megatron-Bridge, а затем проведи DPO-выравнивание.
Оцени хостинговый эндпоинт модели по бенчмарку MCQ и верни результаты.
Настрой параметры для pretraining (CPT) на моем датасете с использованием 8 GPU.

Что понадобится

  • Установленный репозиторий Nemotron с директорией src/nemotron/steps/ и доступ к uv.
  • Файл профиля окружения (NEMOTRON_ENV_FILE или env*.toml) с настройками для удаленного выполнения.
  • Экспортированные переменные окружения для аутентификации (например, NVIDIA_API_KEY), не вложенные в код.

Описание составлено по SKILL.md навыка, сверено 05.10.2026.

Как подключить

1Скачать навык
# возьмите папку навыка «nemotron-customize» из репозитория: # https://github.com/nvidia/skills
2Положить папку с SKILL.md к навыкам ассистента
# положите папку навыка туда, где ассистент ищет skills: cp -r nemotron-customize/ ~/.claude/skills/nemotron-customize/

Навык — папка с файлом SKILL.md (описание + инструкции) и опциональными скриптами. Ассистент (Claude, Claude Code и совместимые) подхватывает его по описанию и следует шагам.