profile-anatomy

Parse TPU pretraining profile data and document its schema.

5|Updated Feb 4, 2026

Search Tags:#schema documentation #pretraining #tpu #profile analysis #xla

Ready to run

npx skills add https://github.com/primatrix/skills --skill profile-anatomy

💬 Prompt for ChatGPT / Claude Chat▼

Please help me install this Agent Skill.
Skill: profile-anatomy
Source: https://github.com/primatrix/skills/tree/main/plugins/tpu-perf/skills/profile-anatomy
Command: npx skills add https://github.com/primatrix/skills --skill profile-anatomy

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill requires xplane_pb2, hlo_pb2, and includes scripts (resource) and references (resource) components.

What problem does it solve?

This Skill provides schema documentation and reference scripts for interpreting TPU pretraining profile data, enabling deeper analysis and development of related tools.

Core Features & Use Cases

Profile Structure: Describes the file layout and schema of TPU pretraining profiles.
Reference Scripts: Provides scripts for parsing and analyzing profile data.
Use Case: When developing or analyzing tools that interact with TPU pretraining profiles, this Skill can help you understand the structure and meaning of the data contained within the profiles.

Quick Start

Use the walk_xplane.py script to explore the full structure of a TPU pretraining profile directory.

Frequently Asked Questions about profile-anatomy

High-intent search queries and answers about installing and using this skill.

FAQPage Schema

How do I parse TPU pretraining profile data for schema documentation?▼

TPU pretraining profiles contain structured performance and trace data. This Skill provides schema documentation and reference scripts to help developers understand the file layout, internal data types, and meaning within xplane.pb and trace.json.gz files.

How do I parse TPU pretraining profile data using reference scripts?▼

You can parse TPU pretraining profile data by running the walk_xplane.py script. This script walks through the profile directory structure, reading xplane.pb and trace.json.gz files to extract and document their internal schema and contents.

Can I use this to analyze XLA profiles generated from TPU pretraining?▼

Yes, you can analyze XLA profiles generated from TPU pretraining. The Skill specifically targets TPU pretraining profile data, providing the necessary schema documentation and parsing scripts to interpret XLA-related structures within the xplane.pb format.

Do I need protobuf dependencies to interpret TPU profile structures?▼

Yes, you need protobuf dependencies to interpret TPU profile structures. The Skill relies on xplane_pb2 and hlo_pb2 protobuf definitions to parse the binary xplane.pb data and extract the underlying schema and trace information.

What is the best way to understand the internal schema of xplane.pb files?▼

The best way to understand the internal schema of xplane.pb files is to use the provided schema documentation alongside the walk_xplane.py script, which systematically traverses and documents the file layout and data types within TPU profiles.