Add Uint Support
pytorch/pytorch
Add unsigned integer (uint) type support to PyTorch operators by updating ATDISPATCH macros.
Learn the target framework from enabled knowledge tools and implement a baseline GPU kernel.
$ npx skills add alibaba/atrex-kernel-agent --skill gpu-kernel-baseline -a claude-codeProject install by default; add -g for ~/.claude/skills/.
$ gh skill install alibaba/atrex-kernel-agent gpu-kernel-baseline --agent claude-codeProject scope by default; add --scope user for a personal install. Needs GitHub CLI 2.90.0 or later (public preview).
$ git clone --depth 1 https://github.com/alibaba/atrex-kernel-agent.git skills-src && mkdir -p .claude/skills && cp -r skills-src/skills/gpu-kernel-baseline .claude/skills/gpu-kernel-baseline && rm -rf skills-srcUse ~/.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/
Install the "gpu-kernel-baseline" agent skill from https://github.com/alibaba/atrex-kernel-agent/tree/main/skills/gpu-kernel-baseline into .claude/skills/gpu-kernel-baseline/ in this project. Copy the whole folder (SKILL.md and every file beside it), keep the folder name "gpu-kernel-baseline", then confirm the skill loads.Claude Code copies the folder itself, the same result as the manual copy. Check what it changed before you commit it.
$skill-installer install https://github.com/alibaba/atrex-kernel-agent/tree/main/skills/gpu-kernel-baselineType this inside Codex. $skill-installer <name> installs a curated skill from openai/skills. The installer writes to $CODEX_HOME/skills (default ~/.codex/skills). Restart Codex if the skill does not show up.
$ npx skills add alibaba/atrex-kernel-agent --skill gpu-kernel-baseline -a codexProject install goes to .agents/skills/; add -g for ~/.codex/skills/.
$ gh skill install alibaba/atrex-kernel-agent gpu-kernel-baseline --agent codexProject scope by default (.agents/skills/); add --scope user for a personal install.
$ git clone --depth 1 https://github.com/alibaba/atrex-kernel-agent.git skills-src && mkdir -p .agents/skills && cp -r skills-src/skills/gpu-kernel-baseline .agents/skills/gpu-kernel-baseline && rm -rf skills-srcUse ~/.agents/skills/ instead of .agents/skills for a personal install.
Codex skills documentation · loads skills from .agents/skills/
Install the "gpu-kernel-baseline" agent skill from https://github.com/alibaba/atrex-kernel-agent/tree/main/skills/gpu-kernel-baseline into .agents/skills/gpu-kernel-baseline/ in this project. Copy the whole folder (SKILL.md and every file beside it), keep the folder name "gpu-kernel-baseline", then confirm the skill loads.Codex copies the folder itself, the same result as the manual copy. Check what it changed before you commit it.
$ npx skills add alibaba/atrex-kernel-agent --skill gpu-kernel-baseline -a cursorProject install goes to .agents/skills/; add -g for ~/.cursor/skills/.
$ gh skill install alibaba/atrex-kernel-agent gpu-kernel-baseline --agent cursorProject scope by default (.agents/skills/); add --scope user for a personal install.
$ git clone --depth 1 https://github.com/alibaba/atrex-kernel-agent.git skills-src && mkdir -p .cursor/skills && cp -r skills-src/skills/gpu-kernel-baseline .cursor/skills/gpu-kernel-baseline && rm -rf skills-srcUse ~/.cursor/skills/ instead of .cursor/skills for a personal install.
Cursor skills documentation · loads skills from .cursor/skills/, .agents/skills/, .claude/skills/, .codex/skills/
Install the "gpu-kernel-baseline" agent skill from https://github.com/alibaba/atrex-kernel-agent/tree/main/skills/gpu-kernel-baseline into .cursor/skills/gpu-kernel-baseline/ in this project. Copy the whole folder (SKILL.md and every file beside it), keep the folder name "gpu-kernel-baseline", then confirm the skill loads.Cursor copies the folder itself, the same result as the manual copy. Check what it changed before you commit it.
$ gemini skills install https://github.com/alibaba/atrex-kernel-agent.git --path skills/gpu-kernel-baseline--scope user (default) or --scope workspace; --path is the subfolder of the repo that holds the skill; --consent skips the security confirmation prompt.
$ npx skills add alibaba/atrex-kernel-agent --skill gpu-kernel-baseline -a gemini-cliProject install goes to .agents/skills/; add -g for ~/.gemini/skills/.
$ gh skill install alibaba/atrex-kernel-agent gpu-kernel-baseline --agent gemini-cliProject scope by default (.agents/skills/); add --scope user for a personal install.
$ git clone --depth 1 https://github.com/alibaba/atrex-kernel-agent.git skills-src && mkdir -p .gemini/skills && cp -r skills-src/skills/gpu-kernel-baseline .gemini/skills/gpu-kernel-baseline && rm -rf skills-srcUse ~/.gemini/skills/ instead of .gemini/skills for a personal install, then run /skills reload.
Gemini CLI skills documentation · loads skills from .gemini/skills/, .agents/skills/
Install the "gpu-kernel-baseline" agent skill from https://github.com/alibaba/atrex-kernel-agent/tree/main/skills/gpu-kernel-baseline into .gemini/skills/gpu-kernel-baseline/ in this project. Copy the whole folder (SKILL.md and every file beside it), keep the folder name "gpu-kernel-baseline", then confirm the skill loads.Gemini CLI copies the folder itself, the same result as the manual copy. Check what it changed before you commit it.
$ gh skill install alibaba/atrex-kernel-agent gpu-kernel-baselineInstalls for Copilot at project scope by default; add --scope user for a personal install. Preview a skill first with gh skill preview. Needs GitHub CLI 2.90.0 or later (public preview).
$ npx skills add alibaba/atrex-kernel-agent --skill gpu-kernel-baseline -a github-copilotProject install goes to .agents/skills/; add -g for ~/.copilot/skills/.
$ git clone --depth 1 https://github.com/alibaba/atrex-kernel-agent.git skills-src && mkdir -p .github/skills && cp -r skills-src/skills/gpu-kernel-baseline .github/skills/gpu-kernel-baseline && rm -rf skills-srcUse ~/.copilot/skills/ instead of .github/skills for a personal install. Commit .github/skills so cloud agent and code review can use it.
GitHub Copilot skills documentation · loads skills from .github/skills/, .claude/skills/, .agents/skills/
Install the "gpu-kernel-baseline" agent skill from https://github.com/alibaba/atrex-kernel-agent/tree/main/skills/gpu-kernel-baseline into .github/skills/gpu-kernel-baseline/ in this project. Copy the whole folder (SKILL.md and every file beside it), keep the folder name "gpu-kernel-baseline", then confirm the skill loads.GitHub Copilot copies the folder itself, the same result as the manual copy. Check what it changed before you commit it.
$ npx skills add alibaba/atrex-kernel-agent --skill gpu-kernel-baseline -a opencodeOpenCode documents no install command of its own. Project install goes to .agents/skills/; add -g for ~/.config/opencode/skills/.
$ gh skill install alibaba/atrex-kernel-agent gpu-kernel-baseline --agent opencodeProject scope by default (.agents/skills/); add --scope user for a personal install.
$ git clone --depth 1 https://github.com/alibaba/atrex-kernel-agent.git skills-src && mkdir -p .opencode/skills && cp -r skills-src/skills/gpu-kernel-baseline .opencode/skills/gpu-kernel-baseline && rm -rf skills-srcUse ~/.config/opencode/skills/ instead of .opencode/skills for a personal install.
OpenCode skills documentation · loads skills from .opencode/skills/, .claude/skills/, .agents/skills/
Install the "gpu-kernel-baseline" agent skill from https://github.com/alibaba/atrex-kernel-agent/tree/main/skills/gpu-kernel-baseline into .opencode/skills/gpu-kernel-baseline/ in this project. Copy the whole folder (SKILL.md and every file beside it), keep the folder name "gpu-kernel-baseline", then confirm the skill loads.OpenCode copies the folder itself, the same result as the manual copy. Check what it changed before you commit it.
gpu-kernel-baselineLearn the target framework from enabled knowledge tools and implement a baseline GPU kernel.
GPU Kernel Baseline is an agent skill from alibaba/atrex-kernel-agent. Learn the target framework from enabled knowledge tools and implement a baseline GPU kernel. Use this skill to understand compute semantics, determine the target platform and framework, search reference implementations, and produce a correct V0 baseline with performance records for later profile-driven optimization.
Its SKILL.md is about 2.1k 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. It works with PyTorch. The repository describes itself as: An end-to-end agent project for GPU kernel implementation, analysis, profiling, and iterative optimization. It helps an agent turn PyTorch logic or an existing kernel into a… The licence is Apache-2.0.
4 steps, taken from the step headings in SKILL.md.
Read from SKILL.md and the folder at commit 3d27c1e. It shows what the files ask for, not the result of running them.
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.
Shell commands in SKILL.md call:
pythongitpython3From the folder's file list and the shell code blocks in SKILL.md.
No URLs in SKILL.md. Its commands use git, which can reach the network depending on how they are called.
From URLs in SKILL.md, links to its own repository left out.
Names no API keys, tokens, secrets or passwords.
From names ending in _API_KEY, _TOKEN, _SECRET, _KEY or _PASSWORD in SKILL.md.
GPU Kernel Baseline loads about 2.1k tokens when it runs. Until then it costs about 84 tokens; SKILL.md has 783 words of instructions outside code blocks.
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.
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.
The full file from alibaba/atrex-kernel-agent at commit 3d27c1e, republished under its Apache-2.0 licence (© alibaba). 783 words, ~2,125 tokens.
.claude/skills/gpu-kernel-baseline/SKILL.md (or your agent's skills folder).Use this skill when the user provides PyTorch logic or a kernel demo and asks to:
kernel.py, reference.py, test_kernel.py, and baseline_report.md for later profile-driven optimization.This stage first understands the PyTorch semantics, then learns the framework APIs (CuteDSL or FlyDSL) through enabled knowledge tools, implements kernel.py and test_kernel.py, validates correctness, records performance, writes baseline_report.md, and writes memory/v0.json.
The orchestrator installs discovered plugin instructions at .atrex_plugins/instructions.md.
kernel_demo.GEMM, Decode Attention, Reduction, or Elementwise.CuteDSLFlyDSLFlyDSL.atrex_plugins/instructions.md and inspect python3 tools/plugin.py list for enabled
tool names and input schemas. Follow the session's phase-specific plugin instructions.plans/v0_plan.md.kernel.py based on PyTorch semantics and the learned framework APIs.Not only must the functionality be correct, but the framework implementation must also be correct, using either CuteDSL or FlyDSL.test_kernel.py using PyTorch logic directly as the correctness reference.ref = pytorch_reference(inputs)
out = kernel_v1(inputs)
rel_err = (out.float() - ref).norm() / ref.norm()
assert rel_err < 0.01rel_err < 0.01; lower precision formats may use task-specific relaxed thresholds.test_kernel.py to prevent hanging:import signal
def timeout_handler(signum, frame):
raise TimeoutError("Test case exceeded timeout limit")
signal.signal(signal.SIGALRM, timeout_handler)
TIMEOUT_SEC = int(os.environ.get("TEST_TIMEOUT_SEC", "30"))
for case in test_cases:
signal.alarm(TIMEOUT_SEC)
try:
run_test(case)
except TimeoutError:
record_failure(case, "TIMEOUT_FAIL")
finally:
signal.alarm(0)--multi-seed and do not launch a separate robustness run for V0:python tools/sandbox.py --kind run --no-sync -- \
python test_kernel.py --version v0 --no-memory Parse the emitted [test_kernel] RESULT_JSON=..., use its performance result and accompanying
correctness status for memory/v0.json, and avoid repeating the expensive baseline workload.
TEST_TIMEOUT_SEC env var).TIMEOUT_FAIL, kill the process, and record the failure in baseline_report.md.rel_err plus PASS/FAIL.latency(us) | TFLOPS | bandwidth(GB/s) | TFLOPS peak utilization(%) | bandwidth peak utilization(%)compute_utilization.py to calculate TFLOPS and bandwidth utilization:python tools/compute_utilization.py --gpu <gpu> --dtype <dtype> --flops-expr '<expr>' --bytes-expr '<expr>' --time-ms <ms> --grid-blocks <blocks>Every theoretical peak, bandwidth, and utilization calculation must cite the auditable spec sources registered in Step 0.
Write baseline_report.md with:
rel_err, PASS/FAIL (include any TIMEOUT_FAIL cases)Write baseline iteration data to memory/v0.json using tools/memory_manager.py:
# Create the iteration file
python tools/memory_manager.py create --workspace kernel_opt_<name> --version v0
# Fill in performance and metadata
python tools/memory_manager.py update --workspace kernel_opt_<name> --version v0 \
--set 'performance.latency_us=<value>' \
--set 'performance.tflops=<value>' \
--set 'performance.bandwidth_gbps=<value>' \
--set 'performance.tflops_peak_utilization_pct=<value>' \
--set 'performance.bandwidth_peak_utilization_pct=<value>' \
--set 'optimization.action_category=baseline' \
--set 'optimization.action_description=<summary>' \
--set 'correctness.rel_err=<value>' \
--set 'correctness.status=PASS' \
--set 'quality_gate.result=PASS'For array fields (pitfalls_and_fixes, references), update the JSON file directly or use read + manual edit + write-back. Fill in:
pitfalls_and_fixes: any errors encountered during implementationreferences: stable source record ids and other docs referenced during learningAfter the quality gate passes, commit:
git add kernel.py test_kernel.py baseline_report.md memory/v0.json README.md
git commit -m "V0: baseline kernel"Each iteration produces a memory/v<N>.json file following the schema defined in reference/v_iteration.schema.json. The JSON structure captures performance data, optimization actions, profile evidence, correctness results, ISA metric progress, search logs, pitfalls and fixes, and references.
Key rules:
masked field defaults to false. When set to true, the file is skipped during reads.README.md and must be derived from documented best practices, hardware specs, and Step 0 Roofline conclusions. Do not fabricate thresholds from experience.kernel.pyreference.pytest_kernel.pybaseline_report.mdmemory/v0.json© alibaba, Apache-2.0. Rendered from Markdown: HTML in the file is shown as text, images as links, and headings moved down two levels. Raw file
Just SKILL.md in skills/gpu-kernel-baseline of alibaba/atrex-kernel-agent.
Open the folder on GitHubat commit 3d27c1e
GPU Kernel Baseline 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.
| Skill | Stars | Used in | Tokens | Auto-check | Licence | Repo updated |
|---|---|---|---|---|---|---|
| GPU Kernel Baseline this skillalibaba/atrex-kernel-agent | 154 | — | ~2.1k | Automated safety check: Pass | Apache-2.0 | |
| Add Uint Supportpytorch/pytorch | 104k | 2 repos | ~2.3k | Automated safety check: Pass | Custom licence | |
| CLIP Image-Text MatchingOrchestra-Research/AI-Research-SKILLs | 13k | 8 repos | ~1.7k | Automated safety check: Pass | MIT | |
| Add Torch Shapes Examplefacebook/pyrefly | 7.1k | — | ~1.3k | Automated safety check: Pass | MIT | |
| Interview Cheatsheetwanshuiyin/ARIS-in-AI-Offer | 574 | 1 repos | ~3.4k | Automated safety check: Notes | MIT | |
| Ghstack CIpytorch/pytorch | 104k | — | ~1.4k | Automated safety check: Pass | Custom licence |
pytorch/pytorch
Add unsigned integer (uint) type support to PyTorch operators by updating ATDISPATCH macros.
Orchestra-Research/AI-Research-SKILLs
Explains OpenAI's CLIP model for zero-shot image classification, image-text similarity, semantic image search and content moderation, with install steps and code patterns.
facebook/pyrefly
A skill your agent uses when adding a new PyTorch model to Pyrefly's shape-tracking example corpus under tensor-shapes/pyrefly-torch-stubs/examples — i.e.
wanshuiyin/ARIS-in-AI-Offer
Generate a long-form Chinese interview-prep cheat sheet on a specific ML/LLM topic — formulas with derivations, from-scratch PyTorch code, comparison tables, and 25 高频面试题 (L1 必会 / L2 进阶 / L3 顶级 lab).
pytorch/pytorch
Manage CI for PyTorch ghstack stacks by running CI where its results are useful now and deferring other PRs with [no-ci].
open-infra-skills/infra-skills
Profiles, benchmarks and tunes AI training workloads on Moore Threads MUSA GPUs with a measurement-first process that keeps model behavior unchanged.
alibaba/atrex-kernel-agent
Mine a per-kernel optimization trace — a git repository capturing successive versions of one kernel being optimized — into structured, gate-validated optimization-experience records for the GPU…
alibaba/atrex-kernel-agent
Mine AI coding-agent session transcripts into structured, gate-validated GPU-kernel optimization records for the wiki.
alibaba/atrex-kernel-agent
Choose and run ACU-only, adaptive PPU in-kernel timeline, or optional bounded joint analysis for a PPU kernel.
alibaba/atrex-kernel-agent
Let AKA autonomously add, run, inspect, and revise intra-kernel timeline probes for standalone CUDA/inline PTX or CuTe DSL when ordinary benchmark, NSYS, or NCU evidence cannot answer a specific…
alibaba/atrex-kernel-agent
Generate a structured implementation plan from an evidence draft.
alibaba/atrex-kernel-agent
Run the evidence loop of one long-horizon GPU kernel optimization episode.
Works with
Categories
Learn the target framework from enabled knowledge tools and implement a baseline GPU kernel. GPU Kernel Baseline is an agent skill from alibaba/atrex-kernel-agent. Learn the target framework from enabled knowledge tools and implement a baseline GPU kernel.
GPU Kernel Baseline fits situations like: understand compute semantics; determine the target platform and framework; search reference implementations; produce a correct V0 baseline with performance records for later profile-driven optimization.
Run `npx skills add alibaba/atrex-kernel-agent --skill gpu-kernel-baseline -a claude-code`. Or copy the skill folder (skills/gpu-kernel-baseline in alibaba/atrex-kernel-agent) into .claude/skills/gpu-kernel-baseline in your project. Claude Code loads it when a task matches its description.
Run `npx skills add alibaba/atrex-kernel-agent --skill gpu-kernel-baseline -a codex`. Or copy the skill folder (skills/gpu-kernel-baseline in alibaba/atrex-kernel-agent) into .agents/skills/gpu-kernel-baseline in your project. Codex loads it when a task matches its description.
Cursor, Gemini CLI, GitHub Copilot and OpenCode also load SKILL.md folders. With the skills CLI, run `npx skills add alibaba/atrex-kernel-agent --skill gpu-kernel-baseline -a cursor` (or -a gemini-cli, github-copilot or opencode for the others). To copy it by hand, put the folder in .cursor/skills/gpu-kernel-baseline, .gemini/skills/gpu-kernel-baseline, .github/skills/gpu-kernel-baseline and .opencode/skills/gpu-kernel-baseline in your project.
Going by SKILL.md and its folder, GPU Kernel Baseline needs the command-line tools its instructions call (python, git and python3). Our summary lists: Python 3.
SKILL.md contains no URLs. Its commands use git, which can reach the network depending on how they are called. This is read from the text; nothing was executed.
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.
GPU Kernel Baseline is published under the Apache-2.0 licence (the repository's licence). It allows redistribution, so the full SKILL.md is shown on this page.
About 2.1k tokens (SKILL.md is roughly 8.5k characters). Agents keep only the skill's name and description in context until a task matches; then they load SKILL.md in full.
Skills that share tags, products or a category with GPU Kernel Baseline: Add Uint Support (pytorch/pytorch, 104k stars), CLIP Image-Text Matching (Orchestra-Research/AI-Research-SKILLs, 13k stars), Add Torch Shapes Example (facebook/pyrefly, 7.1k stars) and Interview Cheatsheet (wanshuiyin/ARIS-in-AI-Offer, 574 stars). The comparison table on this page puts their stars, adoption, token cost, safety result and licence side by side.
alibaba (a GitHub organization) maintains it in alibaba/atrex-kernel-agent, which has 154 GitHub stars. The repository holds 7 skills in this directory. The repository was last updated on September 29, 2026.
Source: alibaba/atrex-kernel-agent on GitHub. Facts on this page come from the repository at the commit we read; the author's words are quoted as theirs.