Mercor
Trainium (NKI) Kernel Expert
- Pay
- $70–90/hr
- Type
- Hourly
- Location
- Remote
- Who can apply
- USA only
- Posted
- Aug 25, 2026
Tip: In our experience, people who apply through a referral link have a better chance of being selected, as I’m also working on Mercor.
Free to apply. Our referral link costs you nothing and doesn’t change your pay. If you’re hired, Mercor may pay us a bonus, which keeps this site free. Never pay anyone to get AI training work.
About this role
Evaluate the quality, correctness, and hardware-appropriateness of Neuron Kernel Interface (NKI) development tasks used to train and evaluate a frontier AI lab's models. You'll assess CUDA→NKI migration fidelity, Trainium-specific performance-optimization quality, and cross-platform numerical-correctness standards — and provide clear, rubric-based written feedback. Basic Qualifications 2+ years of hands-on experience developing or optimizing kernels using the Neuron Kernel Interface (NKI) targeting AWS Trainium/Inferentia2 hardware Strong understanding of NKI-specific development patterns: tile-based computation, SBUF/PSUM/HBM memory-hierarchy management, partition-dimension constraints, and DMA orchestration Demonstrated experience assessing CUDA→NKI migration quality Familiarity with Trainium-specific performance profiling (NeuronCore pipeline utilization, tensor-engine throughput,…
Categories
Similar jobs
-
AI Jailbreak & Prompt-Injection Security Expert New
$50–90/hr
-
Senior Software Engineer, Backend (Node.js, TypeScript, Rust) New
$50–80/hr
-
Senior Software Engineering Experts
$120–180/hr
-
Senior Data Scientist and Data Engineering Experts
$120–180/hr
-
Certified Inpatient Medical Coder
$50–70/hr
-
Rust Architect / Backend Developer