neuron-nki-profiling

Profile NKI kernels on AWS Neuron hardware and extract JSON metrics and traces.

50|4|Updated Apr 23, 2026
One-click install
npx skills add https://github.com/aws-neuron/neuron-agentic-development --skill neuron-nki-profiling
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: neuron-nki-profiling
Source: https://github.com/aws-neuron/neuron-agentic-development/tree/main/skills/neuron-nki-profiling
Command: npx skills add https://github.com/aws-neuron/neuron-agentic-development --skill neuron-nki-profiling

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill includes references (resource) components.

What problem does it solve?

This Skill simplifies profiling and performance analysis of NKI kernels on AWS Neuron hardware.

Core Features & Use Cases

  • Performance Profiling: Generate execution traces and metrics for NKI kernels using Neuron profiling tools.
  • Optimization Insights: Analyze latency, engine utilization, and memory traffic to improve kernel efficiency.
  • Use Case: A developer benchmarking different kernel implementations can compare profiling results to identify bottlenecks and optimize performance.

Quick Start

Use the profiling skill to generate performance metrics by running the profiling commands in your environment after executing your kernel.

Frequently Asked Questions about neuron-nki-profiling

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I profile NKI kernels on AWS Neuron hardware?

Profile NKI kernels on AWS Neuron hardware by executing the kernel in your environment and then running Neuron profiling commands to generate execution traces and metrics for analysis.

What performance metrics can I extract from Neuron kernel profiling?

Neuron kernel profiling extracts performance metrics including latency, engine utilization, and memory traffic using JSON-based metrics extraction to help identify bottlenecks and improve kernel efficiency.

Can I compare different versions of NKI kernels to identify performance bottlenecks?

You can compare different NKI kernel versions by benchmarking their profiling results side by side, analyzing the generated execution traces and metrics to identify performance bottlenecks across implementations.

How do I visualize execution traces for NKI kernels running on Trainium or Inferentia?

Visualize execution traces for NKI kernels on Trainium or Inferentia by running Neuron profiling tools after kernel execution, which generates detailed trace visualizations for debugging and performance tuning.

What do I need to start performance tuning for machine learning kernels on embedded hardware?

Performance tuning for machine learning kernels on embedded hardware requires executing your NKI kernel in a Neuron environment, then running the profiling commands to systematically collect runtime metrics and traces.