Trainium NKI Kernel Expert
Compensation
$70 – $90 /yr
Employment type
Full-time
Work setting
On-site
Location
San Francisco, CA
Schedule
Day shift
Posted
You'll be redirected to the employer's application page.
Job overview
The Trainium NKI Kernel Expert position is remote, with the posting describing the work as remote and compensation of $70–$90 per hour. Mercor connects technical talent with AI research labs. The expert evaluates NKI development tasks for correctness and hardware suitability, reviews CUDA-to-NKI migration fidelity and Trainium performance optimization, and provides rubric-based feedback on numerical correctness. The role collaborates asynchronously with AI research teams to improve AI model performance and dataset coverage.
What you'll do
- Evaluate NKI development tasks for quality, correctness, and hardware appropriateness
- assess CUDA-to-NKI migration fidelity and Trainium performance optimization
- review numerical-correctness standards and provide rubric-based feedback
- collaborate with AI research teams on dataset consistency and coverage.
What we're looking for
- Skills & competencies
- trainiumnkikernel optimizationaws trainiuminferentia2cuda-to-nki migrationnumerical correctnessai modelsremotemercorcontract role
- Work arrangement
- Weekend coverage required
Why this role
Remote work; requires at least two years developing or optimizing NKI kernels for AWS Trainium/Inferentia2 hardware.
About the employer
Mercor is hiring for this role. Industry: Custom Computer Programming Services. Sector: 54.
Additional details
- Industry sector
- 54
- Industry
- Custom Computer Programming Services
- Occupation code
- 15-1252.00
You'll be redirected to the employer's application page.
Listing ID: 7020622a-da5c-4fec-8e60-289decbb9a8d