Logo for Mercor

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

Mercor

Role overview

Qualifications

  • 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.

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.

Key facts

Hard skills

Nvidia CUDA Performance Improvement Memory Management AWS SDK Benchmarking

Other skills

  • Collaboration

About the company

Mercor logo

Mercor

Job Boards & Talent Marketplaces

Our vast talent network trains frontier AI models in the same way teachers teach students: by sharing knowledge, experience, and context that can't be captured in code alone. Today, more than 30,000 experts in our network collectively earn over $2 million a day.

Company details

IndustryJob Boards & Talent Marketplaces
Company size51 - 200

Links

Website LinkedIn See all jobs

Your match analysis

See how your profile stacks up against this role.

We compared the job requirements to your profile to show where you're strong and where you fall short.

Job description

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: https://talent.docs.mercor.com/welcome
  • For any help or support, reach out to: support@mercor.com

PS: Our team reviews applications daily. Please complete your AI interview and application steps to be considered for this opportunity.

Apply once. Then go straight to the hiring manager.

After you apply, unlock the direct contact details of the people who actually make the call. A quick follow-up makes you 5x more likely to land an interview.

MR

Marcus Rivera

Chief Revenue Officer

m.rivera@company.com
linkedin.com/in/marcusrivera
Unlocked after you apply
·

HR Specialist Related jobs

Other jobs at Mercor

Premium

Reach out to the hiring manager directly.

Gain access to the contact details of the hiring managers who actually decide, and reach out to network with them directly. That, plus more when you upgrade:

  • Full match report with fit score and gaps
  • Career diagnostics on how recruiters read you
  • Curated company matches and warm intros
  • 48h early access to new roles

Cancel anytime.