Skills
.Work. You
Rest

agent-profiling

Official

Optimize JAX training performance.

Software Engineering#performance#optimization#profiling#jax#bottleneck analysis#perfetto#xprof
Authormarin-community
Version1.0.0
Installs0

System Documentation

What problem does it solve?

This Skill streamlines the process of identifying and resolving performance bottlenecks in JAX training runs, turning raw profiling data into actionable optimization strategies.

Core Features & Use Cases

  • Profile Ingestion: Parses jax_profile artifacts (xprof/Perfetto traces) into a structured summary.
  • Hotspot Analysis: Identifies performance bottlenecks and compute/communication hotspots.
  • Optimization Workflow: Guides users through a measure-change-re-measure-compare cycle for iterative performance tuning.
  • Use Case: When a JAX training job is running slower than expected, use this skill to analyze the profiling data, pinpoint the slowest operations or communication bottlenecks, and suggest specific code or configuration changes to improve throughput.

Quick Start

Use the agent-profiling skill to analyze the latest profile artifact from the run target 'marin-community/marin/my-training-run-id'.

Dependency Matrix

Required Modules

None required

Components

scriptsreferences

💻 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: agent-profiling
Download link: https://github.com/marin-community/marin/archive/main.zip#agent-profiling

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 223,000+ vetted skills library on demand.