Trainium Kernel Expert – AI Trainer

Job overview

The Trainium Kernel Expert – AI Trainer role is remote, with the listing identifying Mercor as headquartered in San Francisco, CA. Compensation is $70–$90 per hour. Mercor connects creative and technical talent with leading AI research labs. The expert evaluates kernel-development work for correctness, hardware suitability, migration fidelity, and performance. The role provides rubric-based feedback and collaborates with AI research teams to improve consistency and coverage across datasets, contributing to AI model performance.

What you'll do

• Evaluate the quality, correctness, and hardware appropriateness of Neuron Kernel Interface (NKI) development tasks for AI model training and evaluation
• assess CUDA-to-NKI migration fidelity and Trainium-specific optimization
• review cross-platform numerical-correctness standards and provide rubric-based feedback
• collaborate with AI research teams on dataset consistency and coverage
• work independently and asynchronously.

What we're looking for

Skills & competencies

trainium kernel expert

ai trainer

nki

aws trainium

inferentia2

cuda-to-nki migration

kernel optimization

numerical correctness

ai research

remote

mercor

$70–$90 per hour

contract

Work arrangement

• Weekend coverage required

Why this role

Requires 2+ years developing or optimizing NKI kernels for AWS Trainium/Inferentia2 hardware; contract engagement; fully remote.

About the employer

Mercor is hiring for this role. Industry: All Other Professional, Scientific, and Technical Services. Sector: 54.