Skills
.Work. You
Rest

xprof-profiling-analysis

Official

Deep dive into TPU/GPU profiling performance with XProf tools

Data & Analytics#machine learning#profiling#performance analysis#TPU/GPU#XProf
Authorprimatrix
Version1.0.0
Installs0

System Documentation

What problem does it solve?

This Skill allows for detailed analysis of TPU/GPU training performance using XProf MCP tools, providing insights into operator breakdowns, memory usage, and performance comparisons.

Core Features & Use Cases

  • Performance Analysis: Utilizes XProf MCP tools to perform live queries and deep domain analysis for trace parsing and interpretation.
  • Framework Op Breakdown: Provides detailed JAX-level operation statistics with the ability to filter by operation category.
  • Memory Analysis: Analyzes HBM usage, fragmentation, and buffer allocations to identify memory issues.
  • Quick Recipes: Offers concise code snippets for common analysis tasks.
  • Offline CLI Tool: Utilizes xprof.py for additional deep analysis when MCP data is insufficient.
  • Use Case: A user may run this skill after training a machine learning model to diagnose bottlenecks, optimize memory usage, and enhance overall performance.

Quick Start

Run the xprof_profiling_analysis skill with the latest profiling run to get an overview of MXU utilization and identify key performance metrics.

Dependency Matrix

Required Modules

None required

Components

scriptsreferencesassets

💻 Claude Code Installation

Recommended: Let Claude install automatically. Simply copy and paste the text below to Claude Code.

Please help me install this Skill:
Name: xprof-profiling-analysis
Download link: https://github.com/primatrix/skills/archive/main.zip#xprof-profiling-analysis

Please download this .zip file, extract it, and install it in the .claude/skills/ directory.
View Source Repository

Agent Skills Search Helper

Install a tiny helper to your Agent, search and equip skill from 620,000+ vetted skills library on demand.