Land a Remote Job
Mercor · Remote job

Trainium (NKI) Kernel Expert

STEP 1 · PRACTICE FIRSTTake a free AI practice interview for this kind of roleGet my practice interview ↓
STEP 2 · APPLYApply on Mercor when you’re readyApply on Mercor →

You apply directly on Mercor's site. Applying is free. Posted Aug 25, 2026.

Get alerts for new jobs onTelegram
STEP 1 · PRACTICE BEFORE YOUR REAL INTERVIEW

Get your free practice interview for this role

Mercor screens applicants online, often with an AI interview. micro1 offers a free, 18-minute AI practice interview with feedback, and we picked the one closest to this role: Back-End Developer. It covers:

Database ManagementRESTful API DesignAsynchronous Programming

Here’s your practice interview

Allow your microphone when it asks, find a quiet spot, and give yourself about 20 minutes. We’ve also emailed you the link and the free ebook.

Step 2: when you feel ready, apply.

Apply on Mercor →

The practice interview is free and run by micro1. Land Remote Jobs is an independent job board, not part of micro1, Mercor or Turing. Practising helps you prepare; it can’t guarantee the result of your real interview. Want more tips? Read our guide to acing the remote job interview.

STEP 2 · APPLY

Apply for this role

You apply on Mercor's own site. It’s free, and you can apply even if you skip the practice.

Apply on Mercor →

Who this role suits

Working engineers who want flexible, project-based remote work that uses their coding and review skills. A strong portfolio or GitHub helps.

Full job description

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, memory-bandwidth bottlenecks) • Experience defining or evaluating cross-platform numerical-correctness standards (GPU vs Trainium accumulation order, rounding behavior, mixed-precision semantics)

Preferred Qualifications • Direct experience with AWS Neuron SDK, Neuron Compiler internals, or contributions to NKI kernel libraries • Prior CUDA or Triton kernel development • Familiarity with Trainium hardware specifications (NeuronCore-v2 architecture, on-chip SRAM topology, supported data types: FP32/BF16/FP8/INT8) • Experience benchmarking ML training workloads on Trn1/Trn2 instances

Last checked open on Oct 28, 2026. Job description provided by Mercor.

Something not working? Tell us

If the practice interview or the apply link doesn’t open, or the job looks closed, let us know and we’ll fix it.

← Browse all remote jobs