📋 Megatron Memory Estimator
Estimate GPU memory usage for Megatron-based MoE (Mixture of Experts) and dense models. Use when users need to (1) estimate memory from HuggingFace model configs (DeepSeek-V3, Qwen, etc.), (2) plan GPU resource allocation for training, (3) compare different parallelism strategies (TP/PP/EP/CP), (4)
npx playbooks add skill yzlnew/infra-skills --skill megatron-memory-estimator
About Megatron Memory Estimator
Megatron Memory Estimator is a free developer workflow skill that configures AI coding agents to estimate gpu memory usage for megatron-based moe (mixture of experts) and dense models. use when users need to (1) estimate memory from huggingface model configs (deepseek-v3, qwen, etc.), (2) plan gpu resource allocation for training, (3) compare different parallelism strategies (tp/pp/ep/cp), (4) .
Its 43-word system prompt specializes your agent in developer workflow with structured methodology and proven output formats. Install with one command to activate immediately.
Use Cases
- Streamlining git workflows and commit conventions
- Setting up CI/CD pipelines and deployment scripts
- Managing monorepos and multi-package projects
- Automating release notes and changelogs
Example Prompts
System Prompt (43 words)
Estimate GPU memory usage for Megatron-based MoE (Mixture of Experts) and dense models. Use when users need to (1) estimate memory from HuggingFace model configs (DeepSeek-V3, Qwen, etc.), (2) plan GPU resource allocation for training, (3) compare different parallelism strategies (TP/PP/EP/CP), (4)
Frequently Asked Questions
What is Megatron Memory Estimator?
Megatron Memory Estimator is a free developer workflow skill for AI coding agents. Estimate GPU memory usage for Megatron-based MoE (Mixture of Experts) and dense models. Use when users need to (1) estimate memory from HuggingFace model configs (DeepSeek-V3, Qwen, etc.), (2) plan GPU resource allocation for training, (3) compare different parallelism strategies (TP/PP/EP/CP), (4) . It provides a specialized system prompt that configures your agent with developer workflow expertise.
How do I use Megatron Memory Estimator with Claude Code?
Run npx playbooks add skill yzlnew/infra-skills --skill megatron-memory-estimator in your terminal to install Megatron Memory Estimator into your Claude Code session. It works immediately after installation.
Which AI coding agents work with Megatron Memory Estimator?
Megatron Memory Estimator is compatible with Claude Code, Cursor, GitHub Copilot, Windsurf, OpenClaw, Cline, and any AI agent that supports custom system prompts or .cursorrules files.
Is Megatron Memory Estimator free to use?
Yes, Megatron Memory Estimator is completely free and open source. The full source is available on GitHub at https://github.com/yzlnew/infra-skills/tree/main/megatron-memory-estimator. You only need a subscription to the AI agent you use it with.
