PyTorch's skills help an AI agent debug AOTInductor crashes and device mismatch errors, manage ghstack PR stacks and CI runs, and add unsigned integer dtype support to ATen operators. Its other skills cover writing Metal/MPS kernels for Apple Silicon, triaging and minimizing GitHub issue reproductions, adding new operator overloads to native_functions.yaml, and deprecating or removing APIs. Taken together, the skills center on core PyTorch internals and the development workflow around them.
Pytorch
20 skills · 105,211 stars · NOASSERTION
Add Op Overload Checklist for adding a new overload (e.g.
Add Uint Support Add unsigned integer (uint) type support to PyTorch operators by updating AT_DISPATCH macros.
Aoti Debug Debug AOTInductor (AOTI) errors and crashes.
At Dispatch V2 Convert PyTorch AT_DISPATCH macros to AT_DISPATCH_V2 format in ATen C++ code.
Ci Metrics Query PyTorch CI, GitHub Actions, HUD, Grafana, and infrastructure metrics.
Cuda Index Width Choose 32-bit vs 64-bit index math in PyTorch CUDA kernels.
Cuda Streams Manage tensor lifetimes across CUDA streams and decide whether record_stream is avoidable.
Deprecate Feature Deprecate, freeze, or remove PyTorch APIs; choose a removal version, handle won't-fix frozen comp…
Distributed Triage Sub-triages issues in the oncall:distributed queue by assigning distributed module labels, routin…
Docstring Write docstrings for PyTorch functions and methods following PyTorch conventions.
Document Public Apis Document undocumented public APIs in PyTorch by removing functions from coverage_ignore_functions…
Fix Issue Fix bugs reported in PyTorch GitHub issues by reproducing, root-causing, and implementing a fix i…
Ghstack Ci Manage CI for PyTorch ghstack stacks by running CI where its results are useful now and deferring…
Metal Kernel Write Metal/MPS kernels for PyTorch operators.
Pr Review Review PyTorch pull requests for code quality, test coverage, security, and backward compatibility.
Pt2 Bug Basher Debug PyTorch 2 compiler stack failures including Dynamo graph breaks, Inductor codegen errors, A…
Pyrefly Type Coverage Migrate a file to use stricter Pyrefly type checking with annotations required for all functions,…
Scrub Issue Fetch, analyze, reproduce, and minimize GitHub issue reproductions.
Skill Writer Guide users through creating Agent Skills for Claude Code.
Triaging Issues Triages GitHub issues by routing to oncall teams, applying labels, and closing questions.