Trainium (NKI) Kernel Expert

Posted 25 Aug 2026

$70–90/hr

Apply

Platform

Mercor

Work

Remote · Contract · 40 hrs/wk · Remote — United States

Experience

2+ years

Eligibility

Open to United States

Field

Software engineering

Skills

Neuron Kernel Interface (NKI)AWS TrainiumAWS Inferentia2CUDATritonGPU Kernel DevelopmentKernel OptimizationAWS Neuron SDKNeuron CompilerTile-Based ComputationSBUF/PSUM/HBM Memory ManagementDMA OrchestrationTrainium Performance ProfilingCUDA-to-NKI MigrationNumerical CorrectnessMixed-Precision SemanticsML Workload Benchmarking

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…

Read the full description on Mercor

Disclosure: Skillora is independent of Mercor. The apply button carries our referral code, and if you are placed, Mercor pays Skillora a fee. It does not change what you are paid. If Mercor asks you to confirm a referrer, it is Skillora’s founder.

All AI training jobs