Trainium (NKI) Kernel Expert Job at Mercor, Remote

YWxNeFBmUndvc29RbGJFckNhM2ErK2hN
  • Mercor
  • Remote

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

Job Tags

Contract work

Similar Jobs

Hunt St

Senior NodeJS Developer Job at Hunt St

 ...remote talent in the Philippines. You will be hired directly by the client, typically as an independent contractor. We are not an outsourcing agency. All of our roles are 100% remote so you'll be able to work from home. Who The Client Is: The client is a financial... 

Leif Johnson Ford

Porter/Rental Agent Job at Leif Johnson Ford

 ...immediate opening is for Porter/Rental Agent at our brand-new state-of-the-art facility in the Oak Hill area of Austin, TX. This is an entry level full-time position. We will train. Your main scope of employment initially will be washing the rental units. Simultaneously you... 

United States Army

Maintenance & Repairs Job at United States Army

Construction Equipment Repairer Now Hiring Full and Part Time Positions Job Overview: As a Construction Equipment Repairer, you'll keep the Army's engineer equipment safe and operational by repairing and maintaining trucks, bulldozers, power shovels, and other heavy equipment... 

AuraOne Human Data

Sr. Full-Stack Software Engineer Job at AuraOne Human Data

 ...Sr. Full-Stack Software Engineer is a remote engineering review track for evaluating production code, debugging traces, and developer-facing AI outputs against real-world correctness standards. Reviewers reproduce failures, write the unit test the model should have written... 

Shree Narayani Networking Solutions Pvt Ltd

Senior Microsoft 365 Test Engineer Job at Shree Narayani Networking Solutions Pvt Ltd

 ...This Senior Microsoft 365 Test Engineer job is a 12 Months Contract with a client located in New York, NY (Hybrid 3 Days onsite every...  ...platform functionality, integrations, compliance controls, and user experiences. Lead daily and scheduled regression testing activities...