Dataset · Hugging Face

Genghan/NKIBench

Name: NKIBench
Creator: Genghan
License: apache-2.0
Keywords: language:en, license:apache-2.0, size_categories:n<1K, arxiv:2511.15915, region:us, nki, aws-neuron, trainium, kernel, benchmark, environment

NKIBench NKIBench is a benchmark of AWS Neuron Kernel Interface (NKI) kernels paired with NumPy reference implementations. Each task provides a specification, a ground-truth NumPy forward pass, and an optimized NKI kernel targeting AWS Trainium devices, together with tooling to compile, check numerical correctness, and measure on-device latency. Paper. Dataset structure NKIBench/ ├── seeds/ # YAML task specifications (shape-agnostic templates) ├──… See the full description on the dataset page: https://huggingface.co/datasets/Genghan/NKIBench.

View on Hugging FaceReport an issue

Author / organization	Genghan
License (as declared)	apache-2.0
Downloads	190 (as reported by Hugging Face)
Likes	0
Created	18 April 2026
Last updated at source	20 April 2026

nkiaws-neurontrainiumkernelbenchmarkenvironment

Embed this entry

<iframe src="https://silytics.si/embed/item/91f7207e-7044-4d32-8ea0-71ec82b3c2f5" width="420" height="190" style="border:0" loading="lazy" title="Silytics entry"></iframe>

Free, no tracking. More about feeds and embeds.

Silytics shows public metadata from Hugging Face and links to the original. Information may be incomplete or out of date. Last refreshed 16 October 2026. Licenses and terms of the original apply to the work itself. Not financial, legal or professional advice.