Trainium (NKI) Kernel Expert Job at Mercor, San Francisco, CA

  • Mercor
  • San Francisco, CA

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

Marriott

Barista Job at Marriott

 ...requested by Supervisors. PREFERRED QUALIFICATIONS Education: High school diploma or G.E.D. equivalent. Related Work Experience: No related work experience. Supervisory Experience: No supervisory experience. License or Certification: None At... 

ArchCity Defenders

Summer 2027 Partnerships & Organizing Internship/Practicum Job at ArchCity Defenders

ArchCity Defenders (ACD) is a holistic legal advocacy organization that combats the criminalization...  ...skills. Please note that this internship may require hours that include nights and...  ...or other programs.* We plan to host Summer 2027 interns and practicum students in... 

Affirmed Home Care

Pediatric Registered Nurse & Licensed Practical Nurse (RN & LPN) Job at Affirmed Home Care

 ...Pediatric Registered Nurse & Licensed Practical Nurse (RN & LPN) - Home Care Brooklyn, NY | Ventilator & Tracheostomy Experience Required | Open to RN & LPN Join Affirmed Home Care's Growing Pediatric Division! Affirmed Home Care is seeking compassionate an experienced... 

Sportygroup

Data Analyst (Europe) Job at Sportygroup

 ...About the role As a Data Analyst at Sporty, youll be mining through petabytes of data to provide key insights that drive core business...  ...to provide great internal networking opportunities Interview Process - Remote video screening with our Talent Acquisition Team - Offline... 

Bladenboro Family Dentistry

Dental Assistant II Job at Bladenboro Family Dentistry

 ...searching for a highly motivated and compassionate Dental Assistant II. The current opening is for either a permanent full-time or part-time position. Full-time position would be...  ...patient care through a great patient experience. We also promote a team-oriented working...