Port a HuggingFace model to NeuronX Distributed Inference for AWS Trainium/Inferentia. Use when the user says "port model", "autoport", "convert to neuron", "compile for trainium", or invokes /neuron-framework-autoport. Handles the full workflow: knowledge base analysis, architecture analysis, NeuronX implementation, compilation, inference testing, and accuracy validation (95%+ token match).
Verifies functional equivalence between two implementations of the same model using a hierarchical 8-stage algorithm. Orchestrates model tree comparison, component-level R-ratio testing, E2E logit comparison, and distributional checks. Use when porting models between frameworks, hardware targets, or precision regimes โ e.g., HuggingFace to NxDI, PyTorch to ONNX, GPU to Neuron, FP32 to BF16, or any source-target pair.
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.
Guide for writing and modifying NKI kernels. Covers new kernel creation from PyTorch/NumPy/natural language, editing existing kernels, adding shape/dtype support, refactoring tiling strategies, and implementing new features in NKI code. Use when user says "write NKI kernel", "convert PyTorch to NKI", "translate numpy to NKI", "create NKI kernel", "implement in NKI", "NKI version of", "how to write NKI kernel", "add support for <shape/dtype>", "modify this NKI kernel", "extend kernel to handle", "refactor tiling", "change tile size", "add batch dimension", "support variable length", "fix this kernel logic", "update kernel for gen4", or needs NKI API guidance for kernel changes.
This skill guides using the cli to generate NKI kernel profiles (NEFF + NTFF pairs) to analyze performance on Neuron hardware. Use when encountering "profile kernel", "capture execution trace", "generate NEFF", "get summary-json", or asking "how to profile NKI kernel".