Other open roles
Trainium (NKI) Kernel Expert
Work Expert (WE) is an independent publishing and referral website. We are not a recruiter, hiring manager, agent or employer, and we are not affiliated with or endorsed by Mercor. Applying takes you to the platform's own website, where we may be recorded as the referring source. We may receive a referral fee at no additional cost to you. Read our full affiliate disclosure →
About the 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.
What You'll Do
You're a Good Fit If You
- 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)
- 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)
Role Highlights
Pay & Payout
Independent contractor engagement. Apply via the link below - Mercor handles onboarding and pays weekly via Stripe or Wise.
Common Questions
Opens Mercor in a new tab. You pay nothing; we may earn a referral fee.
Get similar jobs every week
Similar roles
More model evaluation work you may qualify for.
BioInformatics Experts
Complete realistic, multi-step scientific data-analysis tasks in bioinformatics Write original bioinformatics questions where each answer is exact and objective
Atomistic Materials Modeling Expert
Mercor is recruiting Atomistic Materials Modelling Experts with strong backgrounds in computational materials science, quantum mechanical simulations, and molecular modelling to
Oncologist
Mercor is collaborating with a leading AI research lab on a project to advance frontier medical problem-solving in oncology. We are looking for oncologists based in the United