module_bottleneck

A performance-analysis skill that finds the slowest PyTorch modules in recent training steps. PyTorch is a machine-learning software library, and a module is a reusable part of a model or processing pipeline.

In plain words
What is it for?
Use it to rank slow modules, limit analysis to recent steps or a chosen hook stage, and decide whether deeper call-stack or distributed-performance profiling is needed.
Why use it?
It identifies which model components consume the most time over ongoing training without relying only on a one-time detailed profiler recording.

Skill for Claude CodeCodex

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.

agentmods
npx agentmods add skills/deeplink-org/probing/module_bottleneck
Any agent
npx skills add DeepLink-org/probing --skill module_bottleneck
Clone the repo
git clone --depth 1 https://github.com/DeepLink-org/probing

Made for: Claude Code, Codex.

Per session 15 Skills are progressive disclosure: only the name and description are preloaded; the body loads when the skill is used.
When invoked 294 The whole file, excluding the scripts and references it only reads on demand.
Security scan A 0 findings. Scan, not verified.
Origin original 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.

ModelPer sessionOnce invoked
Fable 5 $0.00015 $0.00294
Opus 5 $0.00008 $0.00147
Sonnet 5 $0.00003 $0.00059
Haiku 4.5 $0.00002 $0.00029

Measured 3d ago against content hash 5288d6d302c0, method: parsed. Prices are Anthropic first-party input rates as of 2026-08-30, from the pricing page.

Security

Grade A, and why

module_bottleneck 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 3d ago.

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.

python/probing/bundled_skills/module_bottleneck/SKILL.md · 31 lines

What it actually says

PyTorch module bottleneck

基于 python.torch_trace 的 post-hook duration,找出最近 step 中最耗时的模块。 与 torch.profiler 互补:这是长期采样、模块级、低开销视图。

Parameters

  • recent_steps (integer, default 10): Analyze the last N training steps
  • stage_filter (string, default post forward): Hook stage to aggregate (post forward | post step)
  • 打开 Torch 火焰图 (profiling/torch) 看调用栈分布
  • 分布式变慢 → skill: slow_rank
  • 需要 CPU 栈 → SET probing.pprof.sample_freq=99, profiling/pprof
Files

What ships with it

1 file 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.

  1. 3d ago First seen · 31 lines · 15 tokens per session scan A 5288d6d302c0

Subscribe to this mod's changes

module_bottleneck is a skill published in the GitHub repository DeepLink-org/probing (11 stars, last pushed 4d ago), licensed Apache-2.0. It adds 15 tokens to every session and 294 once invoked, about $0.0001 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.