Trainium Kernel Expert - Fully Remote | Upto $90/hr

About the job Mercor connects elite creative and technical talent with leading AI research labs. Headquartered in San Francisco, our investors include Benchmark , General Catalyst , Peter Thiel , Adam D'Angelo , Larry Summers , and Jack Dorsey . Position: Trainium (NKI) Kernel Expert Type: Contract Compensation: $70–$90/hour Location: Remote Role Responsibilities • Evaluate the quality, correctness, and hardware-appropriateness of Neuron Kernel Interface (NKI) development tasks for training and evaluating AI models. • Assess CUDA→NKI migration fidelity and Trainium-specific performance-optimization quality. • Review cross-platform numerical-correctness standards and provide clear, rubric-based feedback. • Collaborate with AI research teams to ensure consistency and coverage across datasets. • Work independently and asynchronously to improve AI model performance . Qualifications Must-Have • 2+ years of experience developing or optimizing kernels using NKI targeting AWS Trainium/Inferentia2 hardware. • Strong understanding of NKI-specific development patterns and memory-hierarchy management. • Experience assessing CUDA→NKI migration quality. • Familiarity with Trainium-specific performance profiling. • Experience defining or evaluating cross-platform numerical-correctness standards. Preferred • 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. • Experience benchmarking ML training workloads on Trn1/Trn2 instances. Application Process (Takes 20–30 mins to complete) • Upload resume • AI interview based on your resume • Submit form Resources & Support • For details about the interview process and platform information, please check: • For any help or support, reach out to: PS: Our team reviews applications daily. Please complete your AI interview and application steps to be considered for this opportunity. Originally posted on Himalayas