neuron-nki-docs

A documentation lookup guide for AWS Neuron Kernel Interface, the programming interface for writing kernels on AWS Trainium and Inferentia chips. It routes questions to API references, tutorials, architecture notes, optimization guides, or error-code explanations.

In plain words
What is it for?
It is used to look up NKI APIs, learn how to perform a task, understand compiler or runtime errors, review chip architecture, and find optimization guidance.
Why use it?
It avoids searching the documentation blindly and helps connect a question to the relevant part of the Neuron documentation.

Skill for Claude CodeCodex

Install

Getting it into your agent

One page per mod, every tool's command on it. A separate URL per tool would split the same page into five that compete with each other.

agentmods
npx agentmods add skills/aws-neuron/neuron-agentic-development/neuron-nki-docs
Any agent
npx skills add aws-neuron/neuron-agentic-development --skill neuron-nki-docs
Clone the repo
git clone --depth 1 https://github.com/aws-neuron/neuron-agentic-development

Made for: Claude Code, Codex.

Per session 84 Skills are progressive disclosure: only the name and description are preloaded; the body loads when the skill is used.
When invoked 2,060 The whole file, excluding the scripts and references it only reads on demand.
Security scan A 0 findings. Scan, not verified.
Origin original No closer match found in the catalogue.
Token cost

What it costs to keep this loaded

Counted locally with the o200k_base tokenizer, which is exact for GPT models; Claude uses its own tokenizer and its counts differ. Treat this as one consistent yardstick across the catalogue rather than a bill. Prices are per million input tokens.

ModelPer sessionOnce invoked
Fable 5 $0.00084 $0.02060
Opus 5 $0.00042 $0.01030
Sonnet 5 $0.00017 $0.00412
Haiku 4.5 $0.00008 $0.00206

Measured 2d ago against content hash e4b9f68c9664, method: parsed. Prices are Anthropic first-party input rates as of 2026-08-30, from the pricing page.

Security

Grade A, and why

neuron-nki-docs scanned grade A with 0 findings against 26 rules in 11 categories — prompt injection, anti-refusal, data exfiltration, privilege escalation, supply chain, agent snooping, system-prompt leakage, SSRF and excessive agency — measured 2d ago.

The scan reads SKILL.md. This mod also ships 18 executable files (references/downloads/average_pool2d_jax.py, references/downloads/average_pool2d_nki_kernels.py, references/downloads/average_pool2d_torch.py, …), listed below but not scanned — reading those needs a real analyzer, not pattern matching.

A static scan of the body, not an audit. Every finding is printed with the line that produced it so you can judge whether it matters here. A mod is markdown that instructs an agent; that is exactly why what it instructs is worth reading.

Nothing flagged

None of the 26 patterns this scan looks for appear in this file: no shell pipes, no recursive deletes, no credential paths, no hidden text, no instruction-override or anti-refusal phrasing, no agent-config snooping. That is not a guarantee, it is the absence of the things that are checkable.

skills/neuron-nki-docs/SKILL.md · 196 lines

How it starts

The opening of the file, as written. The whole thing — 196 lines — stays where its author put it; the contents beside it link to each section on GitHub.

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 Pattern Start With Then Read
nl.* or nisa.* or API name references/indices/symbol-lookup.md Linked API doc
"how to [task]" references/indices/task-routing.md Linked tutorial
"NCC_*" or error code references/debugging/error-codes-index.md Specific error doc
gen2/gen3/gen4/trn1/trn2/trn3 references/architecture/ Specific arch doc
optimize/profile/performance references/optimization/ Relevant guide
browse/list all docs references/indices/hierarchical-toc.md Navigate 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

Directory Contents
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

Read the full file on GitHub · 196 lines

Files

What ships with it

60 files beside SKILL.md in the same directory: the scripts, references and assets a skill reads on demand. Not counted in the per-session cost; read them before you install if any of them is executable.

Changes

What this file has done since we first saw it

Hashed on every crawl. A supply-chain change to an agent config is a question of when, not whether, so the history is kept rather than the latest state alone.

  1. 2d ago First seen · 196 lines · 84 tokens per session scan A e4b9f68c9664

Subscribe to this mod's changes

neuron-nki-docs is a skill published in the GitHub repository aws-neuron/neuron-agentic-development (56 stars, last pushed 14d ago), licensed Apache-2.0. It adds 84 tokens to every session and 2,060 once invoked, about $0.0004 per session on Opus 5. A static security scan graded it A with 0 findings. No closer match exists in the catalogue, so it is treated as the original; first seen 2026-08-30.

Related

Other skills, from other repositories

add-macos-statusbar

Add a macOS menu bar status indicator for NanoClaw. Shows a bolt icon with a green/red dot indicating whether NanoClaw is running, with Start, Stop, and Restart controls. macOS only.

nanocoai/nanoclaw · 49 tokens

cli-anything-3mf

3MF mesh geometry editor — detect and resize cylindrical holes, repair meshes, compare 3D printing files. Works with BambuStudio and PrusaSlicer 3MF files.

HKUDS/CLI-Anything · 44 tokens

platform-port

Guide porting FastLED to new MCU platforms, including int.h types, clockless drivers, SPI implementations, and platform detection. Use when adding support for a new microcontroller family or board.

FastLED/FastLED · 42 tokens

doca-compress

Use this skill for hands-on DOCA Compress programming on a BlueField DPU, ConnectX NIC, or host with DOCA — enabling compress-deflate, decompress-deflate, decompress-lz4-stream, or decompress-lz4-block tasks on a docacompress context (the hardware supports DEFLATE both directions plus LZ4 decompress; LZ4 encode is NOT…

NVIDIA/skills · 242 tokens

pcbway

PCBWay PCB fabrication and assembly — turnkey/consigned assembly, design rules, ordering workflow. Alternative to JLCPCB for manufacturing. Use with KiCad. Use this skill when the user mentions PCBWay, needs turnkey assembly (PCBWay sources parts by MPN), has parts not available on LCSC, needs assembled boards with…

aklofas/kicad-happy · 119 tokens

anta-validation

Validate Arista EOS network state with structured pass/fail verdicts using ANTA. Use for "is this switch healthy", "did my change break anything", "verify BGP/interfaces/hardware are correct", "run a health check on this device". Read-only. A test for a feature the device does not run reports notapplicable — never a…

automateyournetwork/netclaw · 84 tokens