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

The Dog Stop Katy (TDS-0066)

Professional Dog Trainer Job at The Dog Stop Katy (TDS-0066)

 ...The Dog Stop is looking for a Professional Dog Trainer. In this role, you'll train dogs of all ages and breeds, craft personalized training plans, and deliver outstanding customer service to pet owners. Key Responsibilities: Lead group classes and private lessons.... 

Labcorp

Clinical Laboratory Technologist - Cytogenetics Job at Labcorp

 ...Labcorp seeks a cytogenetics technologist for its Brentwood, Tennessee laboratory to perform tissue culture, karyotyping, FISH processing, and related diagnostic procedures. The role requires a bachelor's degree in a biological science and clinical laboratory experience... 

ViewTrade

Boca Raton, Florida - Brokerage firm - Operations Associate at ViewTrade Job at ViewTrade

 ...office. We are seeking full time candidates ranging from entry level to general knowledge of all aspects of Operations processing...  ...provide excellent customer service for both our national and international clients. Must be team oriented and work well under pressure,... 

Zuven technologies Inc

Full Stack Engineer (IPC) Job at Zuven technologies Inc

 ...Senior Full Stack Engineer Web Full Stack *** There is a preference for Full Stack developers who recently ended with Fidelity in last 6 months. But not a must. The Role and the Team Fidelity Brokerage Technology is looking for an experienced Full Stack developer... 

Pitt Electric ORI, LLC

Electrical Mechanic Job at Pitt Electric ORI, LLC

 ...Pitt Electric has been a well-known member of the electrical industry for 40 years. We strive to bring value to our customers on every job, while providing a healthy, safe, and rewarding environment for our employees, as they work and provide for their families. Job...