neuron-nki-writing

To write custom NKI kernels, use this Skill to generate structured import statements, utilities, and kernel code. It guides you through layout manipulation and tiling techniques to achieve high-performance GPU and TPU kernel execution.

The best way to modify existing NKI kernels is by following structured steps for validation, tiling, and layout changes. This Skill provides patterns and utility functions to efficiently edit kernels for specific hardware configurations.

Yes, you can validate tensor inputs before running kernel operations. You can instruct the AI to generate a kernel template, such as a scaled exponential operation, that includes built-in input validation to ensure correctness.

NKI kernel development requires strict adherence to API standards to reduce development errors. This Skill ensures your custom kernel workflows follow best practices for correctness and compatibility across GPU and TPU environments.

Custom kernel development often has layout manipulation errors due to lack of structured patterns. This Skill simplifies the process by enforcing best practices in tiling and layout manipulation, reducing complexity and development errors.