top of page
< Back

Trainium (NKI) Kernel Expert

$70–$90/h

United States

Role overview

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

Similar jobs

AI Developer Trace Task Auditor · Mercor · $70–$90/h · United States

AWS Serverless & Infrastructure-as-Code Task Auditor · Mercor · $70–$90/h · United States

Application Users - Android Studio on macOS - Code · Mercor · $55–$65/h · Remote

You’ll be redirected to our partner’s application page

AI TRAINING OPPORTUNITY

bottom of page
Sporterer
OPEN OPPORTUNITIES

AI Training Opportunities

Loading opportunities…

Sporterer
Future of work · September 2026

How big is the remote / hybrid opportunity?

Remote work is already huge — but the bigger story may be the hundreds of millions of jobs that could still move beyond geography.

Global remote and hybrid work opportunity infographic

Remote work already connects millions of people with companies they would never have been able to work for a decade ago. But compared with the size of the global labour market, it may still be early.

The interesting number is not only how many people work remotely today — it is how many roles could plausibly become remote or hybrid next.

The chart above starts with roughly 3.3 billion people employed worldwide. Around 300 million jobs are already remote or hybrid, while an additional pool of roughly 250 million roles could potentially move in that direction as technology, workflows and company culture continue to change.

Skills are becoming more portable

This matters because more work can increasingly be matched to expertise rather than location. AI training and evaluation are a good example: companies need lawyers, engineers, designers, editors, researchers, finance professionals and many other specialists to help improve AI systems.

For workers, that creates access to projects outside their local market. For companies, it expands the talent pool from one city or country to the world.

Explore remote opportunities →