AIKraft/Skills/TAO запуск workflow

TAO запуск workflow

Shared launch intake for any TAO workflow or action. Use when the user wants to run TAO AutoML, train, evaluate, infer, export, generate TensorRT engines, or launch DEFT/workflow jobs on an execution platform.

nvidia official Другое

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

Skill обеспечивает безопасный запуск любых TAO-воркфлоу (AutoML, обучение, инференс, экспорт, TensorRT, DEFT) через строгий префлайт-контроль. Он гарантирует, что перед выполнением согласованы платформа, образ контейнера, данные и вычислительные ресурсы, а затем управляет жизненным циклом задачи через унифицированный контракт.

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

  • Запуск задачи AutoML с указанием бюджета, алгоритма и метрик
  • Обучение или инференс модели с выбором платформы (Docker, Slurm, Kubernetes, Brev, Virtualenv)
  • Мониторинг запущенной задачи в чате с периодической отправкой статуса
  • Экспорт модели или генерация движков TensorRT с подтверждением образа

Как работает

  1. Выбрать платформу исполнения и настроить параметры мониторинга (интервал, включение отслеживания)
  2. Пройти префлайт: подтвердить образ контейнера, предоставить необходимые учетные данные, проверить доступность данных и вычислительных ресурсов
  3. Согласовать детали запуска (image, platform, datasets, compute shape, expected runtime) через review
  4. Выполнить submit: создать запись задачи (JOB_ID), при необходимости скопировать данные и запустить нативную команду платформы
  5. Мониторить статус через status (polling), получать логи через logs или отменить задачу через cancel

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

Запусти обучение модели ResNet50 на Slurm с мониторингом каждые 5 минут
Запусти AutoML для датасета из S3, бюджет 2 часа, используй образ по умолчанию
Проверь статус задачи tao_job_123 и покажи последние логи, если есть ошибки

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

  • Установленный TAO Skill Bank (TAO_SKILL_BANK_PATH)
  • Доступ к выбранной платформе (Docker, Slurm, Kubernetes, Brev, Virtualenv)
  • Необходимые учетные данные (credentials) для платформы и модели

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

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

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

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