Agent skill

Neuron Nki Docs

by uw-syfi in uw-syfi/vibesys

Research NKI documentation for API lookups, tutorials, error codes, architecture, and optimization guides.

MITAuto-check passed

Install Neuron Nki Docs

skills CLI
$ npx skills add uw-syfi/vibesys --skill neuron-nki-docs -a claude-code

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

GitHub CLI
$ gh skill install uw-syfi/vibesys neuron-nki-docs --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/uw-syfi/vibesys.git skills-src && mkdir -p .claude/skills && cp -r skills-src/resources/skills/neuron-agentic-development/skills/neuron-nki-docs .claude/skills/neuron-nki-docs && 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
neuron-nki-docs
GitHub stars
108
Token cost
~1.9k tokens
SKILL.md length
713 words
Files
104 (incl. references)
Skills in repo
15
Repo updated
First seen
Licence
MIT

At a glance

Research NKI documentation for API lookups, tutorials, error codes, architecture, and optimization guides.

  • Works in 3 steps: Find relevant files through provided… → Read and analyze document and code → Summarize findings with specific file…
  • User asks what does <API do
  • SKILL.md covers Query Type Routing, Search Strategy, Directory Guide and Quick API Reference, plus 6 more sections
  • Instructions only: no scripts, shell commands, URLs or credentials in SKILL.md

What it does

Neuron Nki Docs is an agent skill from uw-syfi/vibesys. Research NKI documentation for API lookups, tutorials, error codes, architecture, and optimization guides. Use when user asks "what does <API do", "how to do <task", "what is error <code", "NKI <API signature", "find NKI tutorial for <topic", "look up <symbol", or needs any NKI documentation reference.

Its SKILL.md is about 1.9k tokens, which your agent loads only when the skill is triggered. The skill folder holds 107 other files, including reference files (for example `references/architecture/nki_arch_guides.md`, `references/architecture/trainium2_arch.md` and `references/architecture/trainium3_arch.md`).

The repository describes itself as: Can AI Agents Build Bespoke Systems? The licence is MIT.

When your agent uses it

  • User asks what does <API do
  • How to do <task
  • What is error <code
  • NKI <API signature

Example prompts

  • “what does <API do”
  • “how to do <task”
  • “what is error <code”
  • “/neuron-nki-docs”

Requirements

  • Python 3

Workflow steps

3 steps, taken from the first numbered list in SKILL.md.

  1. Find relevant files through provided indices using Glob and Grep
  2. Read and analyze document and code
  3. Summarize findings with specific file references

What it can do on your machine

Read from SKILL.md and the folder at commit 999938a. 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

    No scripts in the folder and no shell commands in SKILL.md (its code samples are 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

Neuron Nki Docs loads about 1.9k tokens when it runs, and up to ~276k if it reads all its reference files. Until then it costs about 81 tokens; SKILL.md has 713 words of instructions outside code blocks.

Always · name and description, kept in context so the agent knows when to use it
~81
When it runs · the whole SKILL.md, loaded when a task matches
~1.9k
With references · SKILL.md plus every file in references/, read only if the agent opens them
~276k

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

The full file from uw-syfi/vibesys at commit 999938a, republished under its MIT licence (© uw-syfi). 713 words, ~1,927 tokens.

Download SKILL.mdSave it as .claude/skills/neuron-nki-docs/SKILL.md (or your agent's skills folder). This skill also uses 103 other files; get the full folder from GitHub.
name
neuron-nki-docs
description
Research NKI documentation for API lookups, tutorials, error codes, architecture, and optimization guides. Use when user asks "what does <API> do", "how to do <task>", "what is error <code>", "NKI <API> signature", "find NKI tutorial for <topic>", "look up <symbol>", or needs any NKI documentation reference.
argument-hint
[query or API name]
context
fork

Research $ARGUMENTS thoroughly:

  1. Find relevant files through provided indices using Glob and Grep
  2. Read and analyze document and code
  3. Summarize findings with specific file references

NKI Documentation Lookup

This skill provides comprehensive access to NKI (Neuron Kernel Interface) documentation for AWS Trainium/Inferentia kernel development. Use this skill to answer questions about NKI APIs, tutorials, hardware architecture, error codes, and optimization techniques.

Query Type Routing

Route queries to the appropriate documentation based on the query pattern:

Query PatternStart WithThen Read
nl.* or nisa.* or API namereferences/indices/symbol-lookup.mdLinked API doc
"how to [task]"references/indices/task-routing.mdLinked tutorial
"NCC_*" or error codereferences/debugging/error-codes-index.mdSpecific error doc
gen2/gen3/gen4/trn1/trn2/trn3references/architecture/Specific arch doc
optimize/profile/performancereferences/optimization/Relevant guide
browse/list all docsreferences/indices/hierarchical-toc.mdNavigate tree

Search Strategy

  1. Start with indices - Use the index files in references/indices/ to find the exact documentation needed
  2. Follow links - Read the linked documentation file for full details
  3. Cross-reference - For complex topics, combine information from multiple sources
Index Files
  • indices/symbol-lookup.md - Alphabetical API function index with direct links
  • indices/task-routing.md - Task-oriented guide ("I want to...")
  • indices/hierarchical-toc.md - Complete documentation tree structure

Directory Guide

DirectoryContents
references/architecture/Hardware architecture guides for Trainium/Inferentia generations
references/debugging/Error codes index and individual error documentation
references/downloads/Valid Python kernel examples (deprecated patterns excluded)
references/indices/Navigation aids and lookup tables
references/optimization/Performance tuning, profiling, migration guides
references/programming/Core NKI concepts, tutorials, and API reference
references/programming/api/Detailed API documentation by category
references/programming/tutorials/Step-by-step kernel implementation tutorials
references/reference/FAQ and release notes

Quick API Reference

Most Common APIs

Data Movement:

  • nki.isa.dma_copy - Load/store data between HBM and SBUF
  • nki.isa.dma_transpose - Transpose during DMA transfer

Tensor Operations:

  • nki.isa.nc_matmul - Matrix multiplication on Tensor Engine
  • nki.isa.tensor_tensor - Element-wise operations
  • nki.isa.tensor_scalar - Broadcast scalar operations
  • nki.isa.tensor_reduce - Reduction along axes

Memory Allocation:

  • nl.ndarray - Create tensor in SBUF
  • nl.zeros - Create zero-initialized tensor

Loop Constructs:

  • range - Standard loop iterator (recommended)
  • nl.affine_range / nl.sequential_range / nl.static_range - Legacy aliases for range (all have identical effect in NKI 0.3.0+)

SPMD:

  • nl.program_id - Get current program index
  • nl.num_programs - Get total program count
Module Aliases

The documentation uses these standard aliases:

python
import nki
import nki.language as nl
import nki.isa as nisa
Activation Function Operators

Important: Symbols like nl.exp, nl.sigmoid, nl.tanh, etc. are op specifiers, not standalone callable functions. They must be passed as the op parameter to nisa.activation():

python
# CORRECT: Use as op parameter
result = nisa.activation(op=nl.exp, data=tensor, scale=scale_tensor, bias=bias_tensor)

# INCORRECT: These are NOT callable functions
# result = nl.exp(tensor)  # This does NOT work

Hardware Constraints

Critical limits to remember when answering questions:

ConstraintLimitNotes
Partition dimension (P)≤ 128First axis of on-chip tensors
PSUM free dimension (F)≤ 512 (gen2/3) / ≤ 4096 (gen4)Second axis in PSUM buffer
SBUF free dimension (F)≤ 32767Second axis in SBUF buffer
MatMul K dimension≤ 2048Contraction dimension per tile
Show full SKILL.md (298 more words)Show less
Hardware Generations
GenerationDevicesKey Features
gen2 (v2)Trn1, Inf2Baseline NKI support
gen3 (v3)Trn2FP8 support, Double FP8 mode
gen4 (v4)Trn3MXFP8/MXFP4, Quad-MX mode

Source NKI Kernel

The Python example files in references/downloads/ use the latest NKI API patterns and can be used as reference implementations:

  • average_pool2d_nki_kernels.py — 2D average pooling kernel
  • matrix_multiplication_nki_kernels.py — Tiled matrix multiplication kernel
  • transpose2d_nki_kernels.py — 2D transpose kernel
  • mamba_nki_kernels.py — Fused Mamba state-space model kernel
  • test_nki_isa_local_gather.py — Local gather ISA test

Note: The fused_mamba tutorial (references/programming/tutorials/fused_mamba.md) still uses Beta 1 syntax (nl.load/nl.store). The optimization concepts (loop reordering, data reuse, tiling for temporal locality) are valid, but do NOT copy its code patterns — use nisa.dma_copy from /neuron-nki-writing for any code generation.

Common Query Examples

API Lookup

Query: "What is nisa.nc_matmul?" → Read indices/symbol-lookup.md → Find link → Read programming/api/api-nki-isa-tensor.md#nki-isa-nc_matmul

Error Code

Query: "What does NCC_EVRF001 mean?" → Read debugging/error-codes-index.md → Read debugging/error-codes/EVRF001.md

Tutorial

Query: "How do I implement matrix multiplication?" → Read indices/task-routing.md → Read programming/tutorials/matrix_multiplication.md

Architecture

Query: "What's different in Trainium3?" → Read architecture/trainium3_arch.md

Optimization

Query: "How do I profile my kernel?" → Read optimization/use-neuron-profile.md

Response Guidelines

When answering NKI documentation queries:

  1. Cite sources - Reference the specific documentation file used
  2. Include signatures - For API queries, include function signatures and key parameters
  3. Note hardware requirements - Mention if APIs are generation-specific (v2/v3/v4)
  4. Link related docs - Point to related tutorials or guides for context
  5. Warn about deprecated patterns - If documentation shows old patterns, note this

Error Code Format

NKI compiler errors follow the pattern NCC_<category><number>:

  • EOOM* - Out of memory errors
  • EVRF* - Verification/validation errors
  • EUOC* - Unsupported operation errors
  • EBVF* - Backend verification failures
  • ESPP* - Shape/type errors

The full index is in debugging/error-codes-index.md with individual files in debugging/error-codes/.

SkillPurpose
/neuron-nki-writingWrite NKI kernels from specifications
/neuron-nki-debuggingDebug compiler errors on device
/neuron-nki-profilingProfile kernel performance
/neuron-nki-profile-queryingQuery and analyze kernel profile data

© uw-syfi, MIT. Rendered from Markdown: HTML in the file is shown as text, images as links, and headings moved down two levels. Raw file

Files

SKILL.md and 103 other files (references) in resources/skills/neuron-agentic-development/skills/neuron-nki-docs of uw-syfi/vibesys.

  • SKILL.md
  • references/architecture/nki_arch_guides.md
  • references/architecture/trainium2_arch.md
  • references/architecture/trainium3_arch.md
  • references/architecture/trainium_inferentia2_arch.md
  • references/debugging/error-codes-index.md
  • references/debugging/error-codes/EARG001.md
  • references/debugging/error-codes/EBVF030.md
  • references/debugging/error-codes/EHCA005.md
  • references/debugging/error-codes/EOOM001.md
  • references/debugging/error-codes/EOOM002.md
  • references/debugging/error-codes/ESFH002.md
  • references/debugging/error-codes/ESPP004.md
  • references/debugging/error-codes/ESPP047.md
  • references/debugging/error-codes/EUOC002.md
  • references/debugging/error-codes/EVRF001.md
  • references/debugging/error-codes/EVRF004.md
  • … and 87 more

Open the folder on GitHubat commit 999938a

Compare with similar skills

Neuron Nki Docs 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.

Neuron Nki Docs compared with similar skills
SkillStarsUsed inTokensAuto-checkLicenceRepo updated
Neuron Nki Docs this skilluw-syfi/vibesys108—~1.9kAutomated safety check: PassMIT
Comment Code Generate A Tutorialgithub/awesome-copilot40k1 repos~348Automated safety check: PassMIT
Coding Agentopenclaw/openclaw392k1 repos~2.7kAutomated safety check: PassMIT
Code Reviewflutter/flutter180k—~1.4kAutomated safety check: PassBSD-3-Clause
Simplify CodeNousResearch/hermes-agent252k—~3.7kAutomated safety check: PassMIT
Architectural Code ReviewDonchitos/Claude-Code-Game-Studios26k—~3.5kAutomated safety check: PassMIT

Similar skills

  • Comment Code Generate A Tutorial

    github/awesome-copilot

    Official

    Transform this Python script into a polished, beginner-friendly project by refactoring the code, adding clear instructional comments, and generating a complete markdown tutorial.

    40k GitHub starsUsed in 1 repo~348 tokens
    DevelopmentAuto-check passed
  • Coding Agent

    openclaw/openclaw

    Delegate coding work to Codex, Claude Code, or OpenCode as background workers; not simple edits or read-only code lookup.

    392k GitHub starsUsed in 1 repo~2.7k tokens
    DevelopmentAuto-check passed
  • Code Review

    flutter/flutter

    Performs a comprehensive, multi-step code review of pull requests or local code changes, using iterative refinement (generation, critique, synthesis) to ensure high-quality, actionable feedback.

    180k GitHub stars~1.4k tokensUpdated today
    DevelopmentAuto-check passed
  • Simplify Code

    NousResearch/hermes-agent

    Parallel 4-agent cleanup of recent code changes. An agent skill from NousResearch/hermes-agent.

    252k GitHub stars~3.7k tokensUpdated yesterday
    DevelopmentAuto-check passed
  • Architectural Code Review

    Donchitos/Claude-Code-Game-Studios

    Performs an architectural review of code against coding standards, SOLID, testability and performance, and reports no verdict when the inputs are missing.

    26k GitHub stars~3.5k tokensUpdated 3 days ago
    DevelopmentAuto-check passed
  • Review Code

    penpot/penpot

    Code review flow — review a diff, PR, or code change, delegating the review to a subagent that follows the code-review-criteria skill.

    61k GitHub stars~1k tokensUpdated yesterday
    DevelopmentAuto-check passed

More from uw-syfi/vibesys

All 15 skills in this repo
  • Neuron Nki Profiling

    uw-syfi/vibesys

    This skill guides using the cli to generate NKI kernel profiles (NEFF + NTFF pairs) to analyze performance on Neuron hardware.

    108 GitHub stars~3.2k tokensUpdated today
    Auto-check passed
  • Neuron Nki Debugging

    uw-syfi/vibesys

    This skill guides debugging NKI compilation errors on Neuron hardware.

    108 GitHub stars~2.7k tokensUpdated today
    Auto-check passed
  • Query and analyze NKI kernel profile data from neuron-explorer parquet files.

    108 GitHub stars~4.3k tokensUpdated today
    Auto-check passed
  • Neuron Nki Writing

    uw-syfi/vibesys

    Guide for writing and modifying NKI kernels. An agent skill from uw-syfi/vibesys.

    108 GitHub stars~5k tokensUpdated today
    Auto-check passed
  • Triage PRs

    uw-syfi/vibesys

    Triage the open pull requests of the VibeSys repository. An agent skill from uw-syfi/vibesys.

    108 GitHub stars~1.2k tokensUpdated today
    Auto-check passed
  • Open PR

    uw-syfi/vibesys

    Prepare and open VibeSys pull requests from local repo changes.

    108 GitHub stars~2.8k tokensUpdated today
    Auto-check passed

Questions about Neuron Nki Docs

What does Neuron Nki Docs do?

Research NKI documentation for API lookups, tutorials, error codes, architecture, and optimization guides. Neuron Nki Docs is an agent skill from uw-syfi/vibesys. Research NKI documentation for API lookups, tutorials, error codes, architecture, and optimization guides.

When should I use Neuron Nki Docs?

Neuron Nki Docs fits situations like: user asks what does <API do; how to do <task; what is error <code; NKI <API signature.

How do I install Neuron Nki Docs in Claude Code?

Run `npx skills add uw-syfi/vibesys --skill neuron-nki-docs -a claude-code`. Or copy the skill folder (resources/skills/neuron-agentic-development/skills/neuron-nki-docs in uw-syfi/vibesys) into .claude/skills/neuron-nki-docs in your project. Claude Code loads it when a task matches its description.

How do I install Neuron Nki Docs in Codex?

Run `npx skills add uw-syfi/vibesys --skill neuron-nki-docs -a codex`. Or copy the skill folder (resources/skills/neuron-agentic-development/skills/neuron-nki-docs in uw-syfi/vibesys) into .agents/skills/neuron-nki-docs in your project. Codex loads it when a task matches its description.

Can I use Neuron Nki Docs 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 uw-syfi/vibesys --skill neuron-nki-docs -a cursor` (or -a gemini-cli, github-copilot or opencode for the others). To copy it by hand, put the folder in .cursor/skills/neuron-nki-docs, .gemini/skills/neuron-nki-docs, .github/skills/neuron-nki-docs and .opencode/skills/neuron-nki-docs in your project.

What does Neuron Nki Docs need to run?

SKILL.md names no scripts, command-line tools or credentials: Neuron Nki Docs is instructions for the agent only. Our summary lists: Python 3.

Does Neuron Nki Docs 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 Neuron Nki Docs 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 Neuron Nki Docs use?

Neuron Nki Docs is published under the MIT licence (the repository's licence). It allows redistribution, so the full SKILL.md is shown on this page.

How many tokens does Neuron Nki Docs use?

About 1.9k tokens (SKILL.md is roughly 7.7k characters). Agents keep only the skill's name and description in context until a task matches; then they load SKILL.md in full. Its references folder adds about 274k tokens, read only when the agent opens those files.

What are the alternatives to Neuron Nki Docs?

Skills that share tags, products or a category with Neuron Nki Docs: Comment Code Generate A Tutorial (github/awesome-copilot, 40k stars), Coding Agent (openclaw/openclaw, 392k stars), Code Review (flutter/flutter, 180k stars) and Simplify Code (NousResearch/hermes-agent, 252k stars). The comparison table on this page puts their stars, adoption, token cost, safety result and licence side by side.

Who maintains Neuron Nki Docs?

uw-syfi (a GitHub organization) maintains it in uw-syfi/vibesys, which has 108 GitHub stars. The repository holds 15 skills in this directory. The repository was last updated on October 11, 2026.

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