neuron-nki-writing

Official

Streamline custom kernel creation and modification.

Authoraws-neuron
Version1.0.0
Installs0

System Documentation

What problem does it solve?

This Skill simplifies the process of writing and customizing NKI kernels, reducing the complexity and time required for kernel development.

Core Features & Use Cases

  • Kernel Writing: Guides users through structuring import statements, utilities, and kernel code following best practices.
  • Kernel Modification: Assists in editing existing kernels with steps for validation, tiling, and layout changes.
  • Use Case: Imagine a user needs to implement a new matrix multiplication kernel optimized for a custom hardware configuration; this Skill provides detailed patterns and utility functions for rapid development.

Quick Start

Instruct the AI to generate a new kernel template that performs a scaled exponential operation on a tensor with input validation.

Dependency Matrix

Required Modules

None required

Components

scriptsreferences

💻 Claude Code Installation

Recommended: Let Claude install automatically. Simply copy and paste the text below to Claude Code.

Please help me install this Skill:
Name: neuron-nki-writing
Download link: https://github.com/aws-neuron/neuron-agentic-development/archive/main.zip#neuron-nki-writing

Please download this .zip file, extract it, and install it in the .claude/skills/ directory.
View Source Repository

Agent Skills Search Helper

Install a tiny helper to your Agent, search and equip skill from 471,000+ vetted skills library on demand.