PyTorch HUD
About
Provides a PyTorch CI/CD analytics API for investigating build failures, analyzing test flakiness, and monitoring performance trends across the PyTorch infrastructure.
Details
- Author
- izaitsevfb
- Repository
- izaitsevfb/claude-pytorch-treehugger
- GitHub stars
- 2
- Categories
- Productivity, Developer Tools, Design, File Management, AI, API, Automation, Infrastructure
- Tags
- #analytics
Jump to
Setting up with Highlight
This MCP is not yet compatible with Highlight’s one-click setup. However, you can still use it with Highlight by following these steps:
- Download and install Highlight from highlightai.com/download
- Navigate to the plugins tab and select "Add Custom Plugin"
-
Configure the plugin with the settings below
Plugin Name
PyTorch HUDCommand (node, npx, python, etc.)npxArguments-
Argument 1
-y -
Argument 2
@highlight/mcp-server
Please refer to the README for specific instructions on how to obtain API keys or other required environment variables.
-
Argument 1
- Enable "Start Automatically" if you want the plugin to start when Highlight launches
From the repository
claude mcp add hud -- $(which uvx) --from "git+https://github.com/izaitsevfb/claude-pytorch-treehugger.git" pytorch-hud
pip install -r requirements.txt
get_commit_summary
Basic commit info without jobs.
get_job_summary
Aggregated job status counts.
get_filtered_jobs
Jobs with filtering by status, workflow, or name.
get_failure_details
Failed jobs with detailed failure info.
get_recent_commit_status
Status for recent commits with job statistics.
download_log_to_file
Download logs to local storage.
extract_log_patterns
Find errors, warnings, etc.
extract_test_results
Parse test execution results.
filter_log_sections
Extract specific log sections.
search_logs
Search across multiple logs.
Claude Desktop / Cursor
Paste into your MCP client config file to install this server.
{
"mcpServers": {
"pytorch hud": {
"env": {},
"args": [
"-y",
"@highlight/mcp-server"
],
"command": "npx"
}
}
}Linux
{
"env": [],
"args": [
"-y",
"@highlight/mcp-server"
],
"command": "npx"
}Macos
{
"env": [],
"args": [
"-y",
"@highlight/mcp-server"
],
"command": "npx"
}Windows
{
"env": [],
"args": [
"/c",
"npx",
"-y",
"@highlight/mcp-server"
],
"command": "cmd"
}PyTorch HUD API with MCP Support
A Python library and MCP server for interacting with the PyTorch HUD API, providing access to CI/CD data, job logs, and analytics.
Overview
This project provides tools for PyTorch CI/CD analytics including:
- Data access for workflows, jobs, and test runs
- Efficient log analysis for large CI logs
- ClickHouse query integration for analytics
- Resource utilization metrics
Usage (for humans)
claude mcp add hud -- $(which uvx) --from "git+https://github.com/izaitsevfb/claude-pytorch-treehugger.git" pytorch-hud
Development
```bash
Sign in to leave a review
Use Google, GitHub, or an email account so ratings stay tied to real people.
No reviews posted yet.
