NeMo mBridge (производительность): CUDA-графы
Validate and use CUDA graph capture in Megatron Bridge, including local full-iteration graphs and Transformer Engine scoped graphs for attention, MLP, and MoE modules.
AInvidia
official
Другое
Что делает навык
Проверка и использование захвата CUDA graph в Megatron Bridge, включая локальные графы полной итерации и графы Transformer Engine с ограниченной областью для модулей attention, MLP и MoE.
Как подключить
1Скачать навык
# возьмите папку навыка «nemo-mbridge-perf-cuda-graphs» из репозитория:
# https://github.com/nvidia/skills
2Положить папку с
SKILL.md к навыкам ассистента# положите папку навыка туда, где ассистент ищет skills:
cp -r nemo-mbridge-perf-cuda-graphs/ ~/.claude/skills/nemo-mbridge-perf-cuda-graphs/
Навык — папка с файлом SKILL.md (описание + инструкции) и опциональными скриптами. Ассистент (Claude, Claude Code и совместимые) подхватывает его по описанию и следует шагам.