🔌

tpu-perf

🔌Plugin

primatrix/skills

What it does

Systematic analysis of TPU pretraining efficiency. profile-anatomy: schema + reference scripts for xplane.pb / trace.json.gz. comm-analysis: per-axis NCCL bus BW, peak ICI utilization, and per-step compute/comm overlap. compute-breakdown: source-line aggregation, layer scoping, non-compute audit, v7x roofline shortfall. memory-profile: HBM peak occupancy, alive-buffer attribution, lifetime/shape/tf_op/parent_jit rollups.

🏪

Part of

primatrix/skills

performancetpuprofilingxplanepretraining

Installation

Add marketplace in Claude Code:

/plugin marketplace add primatrix/skills

Step 2. Install plugin:

/plugin install tpu-perf@primatrix-skills

AddedJul 16, 2026

View on GitHub Back to Plugins

More from this repository10

Skill

Skill

session-recorder🎯Skill

Skill

🔌

exec-remote🔌Plugin

Execute Python scripts on remote GPU/TPU clusters via SkyPilot

🔌

gke-tpu🔌Plugin

Plan and render GKE TPU Jobs and nodepool actions with explicit context/namespace safety

🔌

xprof-profiling-analysis🔌Plugin

TPU/XLA profiling analysis: MCP tools for XProf HTTP API (operator breakdowns, memory profiles, A/B comparisons) + methodology skills for trace parsing, MFU calculation, bottleneck identification

🔌

beaver🔌Plugin

Beaver project management: GitHub-native issue lifecycle with explicit /beaver-xxx commands covering the full development cycle (create, design, decompose, dev, PR, beaver-fix, tracker, focus, setup)

🔌

gcloud-support-case🔌Plugin

Open, list, and inspect Google Cloud support cases (Cloud TPU / GKE / GCS) via the Cloud Support API v2, with guided classification, priority confirmation, and formatted case detail output

🔌

session-recorder🔌Plugin

Records the complete session content and logs it to a daily work directory with a dynamic filename based on the active CLI agent. Use this for automated progress tracking and documentation.

🔌

superpowers🔌Plugin

Core skills library for Claude Code: TDD, debugging, collaboration patterns, and proven techniques