Stars
★ 2.8k
Agent Skill
Active
Apache-2.0
NVIDIA/Megatron-Bridge/perf-memory-tuning
Techniques for reducing peak GPU memory in Megatron Bridge — expandable segments, parallelism resizing, activation recompute, CPU offloading constraints, and common OOM fixes.
Quality score 84/100
Installation
Install this skill (Claude Code)
# Clone and copy the skill into your project
git clone https://github.com/NVIDIA/skills.git
mkdir -p .claude/skills
cp -r skills/skills/Megatron-Bridge/perf-memory-tuning .claude/skills/
# Or for personal use: ~/.claude/skills/Related in Megatron-Bridge
Agent Skill
Active
NVIDIA/Megatron-Bridge/adding-model-support
Guide for adding support for new LLM or VLM models in Megatron-Bridge.
Agent Skill
Active
NVIDIA/Megatron-Bridge/build-and-dependency
Dev environment setup for Megatron Bridge — container-based development, uv package management, lockfile regeneration, adding dependencies, Slurm container usage, and common build...
Agent Skill
Active
NVIDIA/Megatron-Bridge/bump-dependency
Bump a pinned dependency (TransformerEngine, Megatron-LM, NRX, etc.), regenerate the lockfile, open a PR, and drive it to green by attaching a watchdog to the "CICD NeMo" workflow...
Agent Skill
Active
NVIDIA/Megatron-Bridge/cicd
CI/CD reference for Megatron Bridge — pipeline structure, commit and PR workflow, CI failure investigation, and common failure patterns.
Agent Skill
Active
NVIDIA/Megatron-Bridge/linting-and-formatting
Code style and quality rules for Megatron Bridge — ruff configuration, naming conventions, type hints, mypy rules, docstrings, copyright headers, logging, and the code review check...
Agent Skill
Active
NVIDIA/Megatron-Bridge/mlm-bridge-training
Run Megatron-LM (MLM) and Megatron Bridge training with mock or real data.