Agent skill

Metal Kernel

by pytorch in pytorch/pytorch

Write Metal/MPS kernels for PyTorch operators. An agent skill from pytorch/pytorch.

Custom licenceAuto-check passedAI & LLM Engineering

Install Metal Kernel

skills CLI
$ npx skills add pytorch/pytorch --skill metal-kernel -a claude-code

Project install by default; add -g for ~/.claude/skills/.

GitHub CLI
$ gh skill install pytorch/pytorch metal-kernel --agent claude-code

Project scope by default; add --scope user for a personal install. Needs GitHub CLI 2.90.0 or later (public preview).

Manual copy
$ git clone --depth 1 https://github.com/pytorch/pytorch.git skills-src && mkdir -p .claude/skills && cp -r skills-src/.agents/skills/metal-kernel .claude/skills/metal-kernel && rm -rf skills-src

Use ~/.claude/skills/ instead of .claude/skills for a personal install. The folder must contain SKILL.md.

Claude Code skills documentation · loads skills from .claude/skills/

Facts

Skill name
metal-kernel
GitHub stars
104k
Token cost
~4.9k tokens
SKILL.md length
1,508 words
Files
1
Skills in repo
16
Repo updated
First seen
Licence
Custom licence

At a glance

Write Metal/MPS kernels for PyTorch operators. An agent skill from pytorch/pytorch.

  • Works in 4 steps: Update native_functions.yaml → Implement Metal Kernel → Implement Host-Side Stub → …
  • Adding MPS device support to operators
  • SKILL.md covers Overview, Step 1: Update…, Step 2: Implement Metal Kernel and Step 3: Implement Host-Side Stub, plus 6 more sections
  • Calls python

What it does

Metal Kernel is an agent skill from 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 nativefunctions.yaml dispatch, host-side operators, and Metal kernel implementation.

Its SKILL.md is about 4.9k tokens, which your agent loads only when the skill is triggered. It is a single SKILL.md file with no bundled scripts.

It sits in AI & LLM Engineering, covering Deep learning, GPU and accelerator computing and Shaders. It works with PyTorch and CUDA. The repository describes itself as: Tensors and Dynamic neural networks in Python with strong GPU acceleration.

When your agent uses it

  • Adding MPS device support to operators
  • Implementing Metal shaders
  • Porting CUDA kernels to Apple Silicon

Example prompts

  • “/metal-kernel”

Requirements

  • Python 3

Workflow steps

4 steps, taken from the step headings in SKILL.md.

  1. Update native_functions.yaml
  2. Implement Metal Kernel
  3. Implement Host-Side Stub
  4. Compile

What it can do on your machine

Read from SKILL.md and the folder at commit 84af535. It shows what the files ask for, not the result of running them.

  • Tool permissions

    Pre-approves nothing: there is no allowed-tools line, so your agent's usual permission prompts apply.

    From allowed-tools in the SKILL.md frontmatter.

  • Runs code

    Shell commands in SKILL.md call:

    • python

    From the folder's file list and the shell code blocks in SKILL.md.

  • Network

    No URLs in SKILL.md.

    From URLs in SKILL.md, links to its own repository left out.

  • Credentials

    Names no API keys, tokens, secrets or passwords.

    From names ending in _API_KEY, _TOKEN, _SECRET, _KEY or _PASSWORD in SKILL.md.

Context cost

Metal Kernel loads about 4.9k tokens when it runs. Until then it costs about 68 tokens; SKILL.md has 1,508 words of instructions outside code blocks.

Always · name and description, kept in context so the agent knows when to use it
~68
When it runs · the whole SKILL.md, loaded when a task matches
~4.9k

Estimates: characters ÷ 4, the usual rule of thumb; real counts depend on the model's tokenizer. Scripts and assets cost tokens only if the agent reads them.

Safety

Auto-check passed

The automated check found no risky patterns in SKILL.md.

Automated static check — not a guarantee. Review scripts before installing. It scans the text of SKILL.md for risky patterns (piping downloads into a shell, reading credential files, hidden Unicode, destructive commands); files beside SKILL.md are not scanned.

SKILL.md

Its licence (Custom licence) doesn't allow us to republish the file, so here is its outline and opening line. It has 1,508 words (~4,949 tokens).

“This skill guides you through implementing Metal kernels for PyTorch operators on Apple Silicon.”

— opening of SKILL.md by pytorch, Custom licence
name
metal-kernel

Read the full SKILL.md on GitHub

Files

Just SKILL.md in .agents/skills/metal-kernel of pytorch/pytorch.

Open the folder on GitHubat commit 84af535

Compare with similar skills

Metal Kernel next to the 5 skills that share the most tags, products or categories with it. Stars are the repository's; “used in” counts other GitHub owners with a copy.

Metal Kernel compared with similar skills
SkillStarsUsed inTokensAuto-checkLicenceRepo updated
Metal Kernel this skillpytorch/pytorch104k—~4.9kAutomated safety check: PassCustom licence
Graphsignalgraphsignal/graphsignal257—~6.2kAutomated safety check: PassApache-2.0
Pytorch Profile Analysislucifer1004/VeloQ127—~1.3kAutomated safety check: PassMIT
At Dispatch V2intel/torch-xpu-ops1153 repos~2.2kAutomated safety check: PassApache-2.0
Hyperpod Version Checkerawslabs/agent-plugins9121 repos~910Automated safety check: PassApache-2.0
Magpie Kernel Evaluatoramd/skills395—~2.3kAutomated safety check: PassMIT

Similar skills

  • Graphsignal

    graphsignal/graphsignal

    Profile AI inference workloads (vLLM, SGLang, TensorRT-LLM, PyTorch, any GPU application) with the Graphsignal profiler and read the results from its local /signals JSON endpoint.

    257 GitHub stars~6.2k tokensUpdated 9 days ago
    AI & LLM EngineeringAuto-check passed
  • Pytorch Profile Analysis

    lucifer1004/VeloQ

    Analyze single-file PyTorch/Kineto Chrome trace .json(.gz) files using the VeloQ CLI.

    127 GitHub stars~1.3k tokensUpdated 2 days ago
    AI & LLM EngineeringAuto-check passed
  • At Dispatch V2

    intel/torch-xpu-ops

    Official

    Convert PyTorch ATDISPATCH macros to ATDISPATCHV2 format in ATen C++ code.

    115 GitHub starsUsed in 3 repos~2.2k tokens
    AI & LLM EngineeringAuto-check passed
  • Hyperpod Version Checker

    awslabs/agent-plugins

    Official

    Check and compare software component versions on SageMaker HyperPod cluster nodes - NVIDIA drivers, CUDA toolkit, cuDNN, NCCL, EFA, AWS OFI NCCL, GDRCopy, MPI, Neuron SDK (Trainium/Inferentia)…

    912 GitHub starsUsed in 1 repo~910 tokens
    AI & LLM EngineeringAuto-check passed
  • Benchmarks LLM inference and drives GPU kernel optimization with Magpie.

    395 GitHub stars~2.3k tokensUpdated today
    AI & LLM EngineeringAuto-check passed
  • GPU Optimizer

    Mathews-Tom/armory

    GPU optimization for consumer NVIDIA GPUs (8-24GB VRAM) covering mixed precision, gradient checkpointing, XGBoost GPU, CuPy/cuDF migration, and torch.compile.

    327 GitHub stars~3.5k tokensUpdated yesterday
    AI & LLM EngineeringAuto-check: notes

More from pytorch/pytorch

All 16 skills in this repo
  • Add Uint Support

    pytorch/pytorch

    Add unsigned integer (uint) type support to PyTorch operators by updating ATDISPATCH macros.

    104k GitHub starsUsed in 2 repos~2.3k tokens
    Auto-check passed
  • Docstring

    pytorch/pytorch

    Write docstrings for PyTorch functions and methods following PyTorch conventions.

    104k GitHub starsUsed in 2 repos~2.6k tokens
    Auto-check passed
  • Aoti Debug

    pytorch/pytorch

    Debug AOTInductor (AOTI) errors and crashes. An agent skill from pytorch/pytorch.

    104k GitHub starsUsed in 1 repo~1.7k tokens
    Auto-check passed
  • CI Metrics

    pytorch/pytorch

    Query PyTorch CI, GitHub Actions, HUD, Grafana, and infrastructure metrics.

    104k GitHub stars~1.1k tokensUpdated today
    Auto-check passed
  • Cuda Index Width

    pytorch/pytorch

    Choose 32-bit vs 64-bit index math in PyTorch CUDA kernels. An agent skill from pytorch/pytorch.

    104k GitHub stars~1.6k tokensUpdated today
    Auto-check passed
  • Document Public APIs

    pytorch/pytorch

    Document undocumented public APIs in PyTorch by removing functions from coverageignorefunctions and coverageignoreclasses in docs/source/conf.py, running Sphinx coverage, and adding the appropriate…

    104k GitHub stars~4.2k tokensUpdated today
    Auto-check passed

Works with

Questions about Metal Kernel

What does Metal Kernel do?

Write Metal/MPS kernels for PyTorch operators. An agent skill from pytorch/pytorch. Metal Kernel is an agent skill from pytorch/pytorch. Write Metal/MPS kernels for PyTorch operators.

When should I use Metal Kernel?

Metal Kernel fits situations like: adding MPS device support to operators; implementing Metal shaders; porting CUDA kernels to Apple Silicon.

How do I install Metal Kernel in Claude Code?

Run `npx skills add pytorch/pytorch --skill metal-kernel -a claude-code`. Or copy the skill folder (.agents/skills/metal-kernel in pytorch/pytorch) into .claude/skills/metal-kernel in your project. Claude Code loads it when a task matches its description.

How do I install Metal Kernel in Codex?

Run `npx skills add pytorch/pytorch --skill metal-kernel -a codex`. Or copy the skill folder (.agents/skills/metal-kernel in pytorch/pytorch) into .agents/skills/metal-kernel in your project. Codex loads it when a task matches its description.

Can I use Metal Kernel in Cursor, Gemini CLI or GitHub Copilot?

Cursor, Gemini CLI, GitHub Copilot and OpenCode also load SKILL.md folders. With the skills CLI, run `npx skills add pytorch/pytorch --skill metal-kernel -a cursor` (or -a gemini-cli, github-copilot or opencode for the others). To copy it by hand, put the folder in .cursor/skills/metal-kernel, .gemini/skills/metal-kernel, .github/skills/metal-kernel and .opencode/skills/metal-kernel in your project.

What does Metal Kernel need to run?

Going by SKILL.md and its folder, Metal Kernel needs the command-line tools its instructions call (python). Our summary lists: Python 3.

Does Metal Kernel access the network?

SKILL.md contains no URLs. Any network use would come from the scripts or tools the agent runs. This is read from the text; nothing was executed.

Is Metal Kernel safe to install?

Our automated static check of SKILL.md found no risky patterns, such as piping downloads into a shell, reading credential files or hidden Unicode. It is not a guarantee. Review the folder before installing.

What licence does Metal Kernel use?

Metal Kernel has a licence file (the repository's licence) that doesn't match a standard licence. Read it on GitHub before reusing the skill.

How many tokens does Metal Kernel use?

About 4.9k tokens (SKILL.md is roughly 20k characters). Agents keep only the skill's name and description in context until a task matches; then they load SKILL.md in full.

What are the alternatives to Metal Kernel?

Skills that share tags, products or a category with Metal Kernel: Graphsignal (graphsignal/graphsignal, 257 stars), Pytorch Profile Analysis (lucifer1004/VeloQ, 127 stars), At Dispatch V2 (intel/torch-xpu-ops, 115 stars) and Hyperpod Version Checker (awslabs/agent-plugins, 912 stars). The comparison table on this page puts their stars, adoption, token cost, safety result and licence side by side.

Who maintains Metal Kernel?

pytorch (a GitHub organization) maintains it in pytorch/pytorch, which has 103,819 GitHub stars. The repository holds 16 skills in this directory. The repository was last updated on October 7, 2026.

Source: pytorch/pytorch on GitHub. Facts on this page come from the repository at the commit we read; the author's words are quoted as theirs.