What is megatron-memory-estimator?

A guide for estimating the GPU memory needed by Megatron-based language models, including dense models and mixture-of-experts models. A mixture-of-experts model routes each input through selected subnetworks instead of using one network for every input.

Use it to estimate memory from Hugging Face model configurations, plan GPU resources, and compare tensor, pipeline, expert, and context parallelism strategies.

It helps determine whether planned training runs fit available GPUs and how different ways of splitting work affect memory needs.

megatron-memory-estimator — Skill for Claude Code and Codex by yzlnew

Install

Getting it into your agent

One page per mod, every tool's command on it. A separate URL per tool would split the same page into five that compete with each other.

Any agent

npx skills add yzlnew/infra-skills --skill megatron-memory-estimator

Clone the repo

git clone --depth 1 https://github.com/yzlnew/infra-skills

Made for: Claude Code, Codex.

Wrote this? Show the measurements

A badge with what this costs and how it scanned, read live from this page, so it follows the numbers instead of freezing them. Markdown for a README, HTML for a documentation site or a project page.

README.md

[![agentmods](https://agentmods.dev/badge/skills/yzlnew/infra-skills/megatron-memory-estimator/github.svg)](https://agentmods.dev/skills/yzlnew/infra-skills/megatron-memory-estimator)

Your own site

<a href="https://agentmods.dev/skills/yzlnew/infra-skills/megatron-memory-estimator"><img src="https://agentmods.dev/badge/skills/yzlnew/infra-skills/megatron-memory-estimator/github.svg" alt="Measured on agentmods" height="20"></a>

Or the 80×15 button, for a site that already has a row of RSS and ATOM ones. Only the verdict fits; the numbers stay here.

Your own site · 80×15

<a href="https://agentmods.dev/skills/yzlnew/infra-skills/megatron-memory-estimator"><img src="https://agentmods.dev/badge/skills/yzlnew/infra-skills/megatron-memory-estimator.svg" alt="Reviewed on agentmods" width="80" height="20"></a>

Per session 105 Skills are progressive disclosure: only the name and description are preloaded; the body loads when the skill is used.

When invoked 2,418 The whole file, excluding the scripts and references it only reads on demand.

Security scan A 0 findings. A grade says what 26 rules found in the file — not that it is safe.

Origin unknown No closer match found in the catalogue.

Token cost

What it costs to keep this loaded

Counted locally with the o200k_base tokenizer, which is exact for GPT models; Claude uses its own tokenizer and its counts differ. Treat this as one consistent yardstick across the catalogue rather than a bill. Prices are per million input tokens.

Model	Per session	Once invoked
Fable 5.1	$0.00105	$0.02418
Opus 5	$0.00053	$0.01209
Sonnet 5	$0.00021	$0.00484
Haiku 4.5	$0.00011	$0.00242

Measured 11d ago against content hash 379684738513, method: parsed. Prices are Anthropic first-party input rates as of 2026-09-11, from the pricing page.

Security

Grade A, and why

megatron-memory-estimator scanned grade A with 0 findings against 26 rules in 11 categories — prompt injection, anti-refusal, data exfiltration, privilege escalation, supply chain, agent snooping, system-prompt leakage, SSRF and excessive agency — measured 11d ago.

The scan reads SKILL.md. This mod also ships 8 executable files (megatron_memory_estimator/__init__.py, megatron_memory_estimator/estimate_013.py, megatron_memory_estimator/moe_mem_estimator/__init__.py, …), listed below but not scanned — reading those needs a real analyzer, not pattern matching.

A static scan of the body, not an audit. Every finding is printed with the line that produced it so you can judge whether it matters here. A mod is markdown that instructs an agent; that is exactly why what it instructs is worth reading.

Nothing flagged

None of the 26 patterns this scan looks for appear in this file: no shell pipes, no recursive deletes, no credential paths, no hidden text, no instruction-override or anti-refusal phrasing, no agent-config snooping. That is not a guarantee, it is the absence of the things that are checkable.

megatron-memory-estimator/SKILL.md · 277 lines

The source is not reproduced here

No licence file

Read it on GitHub

Files

What ships with it

12 files beside SKILL.md in the same directory: the scripts, references and assets a skill reads on demand. Not counted in the per-session cost; read them before you install if any of them is executable.

Changes

What this file has done since we first saw it

Hashed on every crawl. A supply-chain change to an agent config is a question of when, not whether, so the history is kept rather than the latest state alone.

11d ago First seen · 277 lines · 105 tokens per session scan A 379684738513

Subscribe to this mod's changes

megatron-memory-estimator is a skill published in the GitHub repository yzlnew/infra-skills (145 stars, last pushed 2mo ago), with no licence file. It adds 105 tokens to every session and 2,418 once invoked, about $0.0005 per session on Opus 5. A static security scan graded it A with 0 findings. No closer match exists in the catalogue, so it is treated as the original; first seen 2026-08-30.

Data and AI ai-infra dsh dsh-plugin skills

upstash-vector-js

Work with the @upstash/vector TypeScript/JavaScript SDK, a serverless vector database for embeddings, similarity search, semantic search, and RAG (retrieval-augmented generation). Use when upserting, querying, fetching, ranging, or deleting vectors, upserting raw text against an index with a built-in embedding model…

upstash/skills · 152 tokens

ito-compute

Query live GPU inventory, submit an authenticated Itô fixed-rate RFQ, inspect RFQ or procurement status, revoke device credentials, and run explicitly gated node qualification through the separately installed canonical CLI. Use when a user asks to find H100/H200 capacity, request a fixed compute rate, check Itô…

gongyijie85/dsh-ecc · 90 tokens

ito-inference

Inspect the availability of model serving on a completed Itô compute booking and, when the canonical backend becomes available, hand off an explicitly confirmed serving manifest. Use after ito-compute has booked GPU nodes and the user asks for an OpenAI-compatible endpoint, ito-serve, hosted Kimi, or self-hosted…

gongyijie85/dsh-ecc · 80 tokens

ito-training

Inspect the availability of ML training on a completed Itô compute booking and, when the canonical backend becomes available, hand off an explicitly confirmed training manifest. Use after ito-compute has booked GPU nodes and the user wants pre-training, fine-tuning, or RL on that metal. ECC implements no training…

gongyijie85/dsh-ecc · 68 tokens

cvqnn

Skill for understanding, using, and implementing the Continuous Variable Quantum Neural Network (CVQNN) for binary classification via the CVQNNAlgorithm class, CVSimulator, and CVClassifier. Skill-first for covered code generation, runnable examples, execution, debugging, validation, and fixed workflows.

unitarylab/quantum-practices · 61 tokens

vqc

Skill for understanding, using, and implementing the Variational Quantum Classifier (VQC) for Iris dataset classification with data re-uploading and Parameter Shift Rule via the VQCAlgorithm class. Skill-first for covered code generation, runnable examples, execution, debugging, validation, and fixed workflows.