xprof-cli

vlasenkoalexey/xprof-cli
★ 7 stars Python AI/LLM Updated today
CLI-first TPU/GPU profile analysis for AI agents (xprof/openxla) — roofline, HLO, Pallas kernel/LLO tools; also ships an MCP server
View on GitHub → 🔍 Audit Wallet Slippage →

Quick Install

Copy the config for your editor. Some servers may need additional setup — check the README.

Add to claude_desktop_config.json:

{
  "mcpServers": {
    "xprof-cli": {
      "command": "uvx",
      "args": [
        "xprof-cli"
      ]
    }
  }
}

Or install with pip: pip install xprof-cli

README Excerpt

CLI-first TPU/GPU profile analysis for AI agents and humans — analyze JAX / PyTorch-XLA / TensorFlow profiles via the open-source server** for assistants that prefer structured tools (Claude Code, Gemini, Cursor, etc.). > **Renamed from `xprof-mcp`** (2026-07). Same repo, full history preserved; > GitHub redirects all old URLs, so existing clones, submodules, and MCP

Tools (20)

XLA_HLO_DUMP_DIRXLA_JF_DUMP_DIRXPROF_LOGDIRXPROF_MODEXPROF_URLaggregate_xplane_eventscheck_kernel_profilingdetect_unfused_reshapesdiff_hlo_stagesget_custom_call_mlirget_device_informationget_device_wall_reportget_framework_op_statsget_hlo_dumpget_hlo_dump_neighborhoodget_hlo_module_contentget_hlo_neighborhoodget_hostsget_input_pipelineget_kernel_stage_breakdown

Topics

cligpumcpmodel-context-protocolpallasprofilingtpuxprof