Claude Skills Online Marketplace

Add unsigned integer (uint) type support to PyTorch operators by updating AT_DISPATCH macros. Use when adding support for uint16, uint32, uint64 types to operators, kernels, or...

Add Unsigned Integer (uint) Support to Operators This skill helps add support for unsigned integer types (uint16, uint32, uint64) to PyTorch operators by updating their AT_DISPATCH macros. When to use this skill Use this skill when: Adding uint16, uint32, or uint64 support to an...

by pytorch • pytorch-pytorch

Debug AOTInductor (AOTI) errors and crashes. Use when encountering AOTI segfaults, device mismatch errors, constant loading failures, or runtime errors from aot_compile, aot_load,...

AOTI Debugging Guide This skill helps diagnose and fix common AOTInductor issues. Error Pattern Routing **Check the error message and route to the appropriate sub-guide:** Triton Index Out of Bounds If the error matches this pattern: **→ Follow the guide in ** All Other Errors...

by pytorch • pytorch-pytorch

Convert PyTorch AT_DISPATCH macros to AT_DISPATCH_V2 format in ATen C++ code. Use when porting AT_DISPATCH_ALL_TYPES_AND*, AT_DISPATCH_FLOATING_TYPES*, or other dispatch macros to...

AT_DISPATCH to AT_DISPATCH_V2 Converter This skill helps convert PyTorch's legacy AT_DISPATCH macros to the new AT_DISPATCH_V2 format, as defined in . When to use this skill Use this skill when: Converting AT_DISPATCH_* macros to AT_DISPATCH_V2 Porting ATen kernels to use the...

by pytorch • pytorch-pytorch

Query PyTorch CI, GitHub Actions, HUD, Grafana, and infrastructure metrics. Use when users ask about CI duration, job failures, queue times, workflow trends, runner health,...

PyTorch CI Metrics PyTorch CI and infrastructure metrics are exposed through Grafana. Use for all Grafana access; it configures the PyTorch Grafana server, context, and authentication. Only users with write permission to the repo have access to Grafana. The authentication only...

by pytorch • pytorch-pytorch

Choose 32-bit vs 64-bit index math in PyTorch CUDA kernels. Use when fixing large-tensor indexing overflows, deciding whether to use int64_t, canUse32BitIndexMath,...

CUDA Index Width in PyTorch Use this skill when a CUDA kernel overflows indexing, fails near elements, or needs a review of vs index math. Start with the overflow site Do not blindly convert all index variables to . Find the expression that can exceed 32 bits and classify where...

by pytorch • pytorch-pytorch

Sub-triages issues in the oncall:distributed queue by assigning distributed module labels, routing to sub-oncalls, and marking triaged. Use when an issue has been routed to...

Distributed Issue Triage Sub-Skill This sub-skill picks up where the PT-level triage bot leaves off. It processes issues that already have the label and performs second-level triage: routing to a distributed sub-oncall, classifying by module, and marking triaged. Contents - Step...

by pytorch • pytorch-pytorch

Write docstrings for PyTorch functions and methods following PyTorch conventions. Use when writing or updating docstrings in PyTorch code.

PyTorch Docstring Writing Guide This skill describes how to write docstrings for functions and methods in the PyTorch project, following the conventions in and . General Principles Use **raw strings** ( ) for all docstrings to avoid issues with LaTeX/math backslashes Follow...

by pytorch • pytorch-pytorch

Document undocumented public APIs in PyTorch by removing functions from coverage_ignore_functions and coverage_ignore_classes in docs/source/conf.py, running Sphinx coverage, and...

Document Public APIs This skill documents undocumented public APIs in PyTorch by removing entries from the coverage ignore lists in and adding Sphinx autodoc directives (e.g., , , , ) to the corresponding or doc source files in . **"Documenting" means adding autodoc directives...

by pytorch • pytorch-pytorch

Fix bugs reported in PyTorch GitHub issues by reproducing, root-causing, and implementing a fix in the local working tree. Use when the user asks to fix a PyTorch GitHub issue.

Fix PyTorch GitHub Issue You are The Fixer. Your goal is to fix the bug reported in a PyTorch GitHub issue. You have an obsession with fixing the root cause of issues, and never settle for hacks that work around things. You are a master at debugging and dive deep to understand...

by pytorch • pytorch-pytorch

Write Metal/MPS kernels for PyTorch operators. Use when adding MPS device support to operators, implementing Metal shaders, or porting CUDA kernels to Apple Silicon. Covers...

Metal Kernel Writing Guide This skill guides you through implementing Metal kernels for PyTorch operators on Apple Silicon. **Important:** The goal of this skill is to use native Metal capabilities via the infrastructure, NOT MPSGraph. Native Metal kernels provide better...

by pytorch • pytorch-pytorch

Review PyTorch pull requests for code quality, test coverage, security, and backward compatibility. Use when reviewing PRs, when asked to review code changes, or when the user...

PyTorch PR Review Skill Review PyTorch pull requests focusing on what CI cannot check: code quality, test coverage adequacy, security vulnerabilities, and backward compatibility. Usage Modes No Argument If the user invokes with no arguments, **do not perform a review**. Instead,...

by pytorch • pytorch-pytorch

Debug PyTorch 2 compiler stack failures including Dynamo graph breaks, Inductor codegen errors, AOTAutograd crashes, and accuracy mismatches. Use when encountering torch.compile...

PT2 Bug Basher Debug test failures and runtime errors in the PyTorch 2 compiler stack (Dynamo, Inductor, AOTAutograd, FX graphs). Workflow Summary **Environment check** -- Ask the user which conda environment to use. Verify it is active by checking . Then run to confirm torch is...

by pytorch • pytorch-pytorch

Migrate a file to use stricter Pyrefly type checking with annotations required for all functions, classes, and attributes.

Pyrefly Type Coverage Skill Prerequisites The file must live in a project with a . , , and the project's test runner must be on PATH. **If any are missing, stop and ask whether a conda environment needs activating** — don't install or substitute (per repo CLAUDE.md). Step 1:...

by pytorch • pytorch-pytorch

Fetch, analyze, reproduce, and minimize GitHub issue reproductions. Use when asked to check if an issue reproduces, minimize a repro, analyze a bug report, or scrub/triage an...

Minimize Issue Reproduction Fetch a GitHub issue, evaluate whether it has a reasonable repro, check if it still reproduces, and systematically minimize the repro to the smallest possible self-contained script. Tools Assume the current environment is correct and run directly....

by pytorch • pytorch-pytorch

Guide users through creating Agent Skills for Claude Code. Use when the user wants to create, write, author, or design a new Skill, or needs help with SKILL.md files, frontmatter,...

Skill Writer This Skill helps you create well-structured Agent Skills for Claude Code that follow best practices and validation requirements. When to use this Skill Use this Skill when: Creating a new Agent Skill Writing or updating SKILL.md files Designing skill structure and...

by pytorch • pytorch-pytorch

Triages GitHub issues by routing to oncall teams, applying labels, and closing questions. Use when processing new PyTorch issues or when asked to triage an issue.

PyTorch Issue Triage Skill This skill helps triage GitHub issues by routing issues, applying labels, and leaving first-line responses. Contents - Step 0: Already Routed — SKIP Step 1: Question vs Bug/Feature Step 1.5: Needs Reproduction — External Files Step 2: Transfer Step...

by pytorch • pytorch-pytorch