Trainium (NKI) Kernel Expert
Recruitment Room →Trainium (NKI) Kernel Expert at Recruitment Room is a contract role based in Remote · United States-231 (remote). The listing states pay of $145,600–$187,200 and asks for 2–15 yrs of experience. Listed skills: CUDA, AWS and SDK. It was published on 27 August 2026 and was open at last check.
| Role | Trainium (NKI) Kernel Expert |
|---|---|
| Company | Recruitment Room |
| Location | Remote · United States-231 (remote) |
| Employment type | Contract |
| Pay (as listed) | $145,600–$187,200 |
| Experience asked | 2–15 yrs |
| Skills listed | CUDA, AWS and SDK |
| Published | 27 August 2026 |
| Status | Open at last check |
Trainium (NKI) Kernel Expert
Remote | Independent Contractor | United States | $145,600–$187,200 annualized ($70–$90/hour)
About the Role
Apply your specialized AWS Trainium and Neuron Kernel Interface (NKI) expertise to help improve the quality and reliability of advanced AI systems.
As a Trainium (NKI) Kernel Expert, you’ll evaluate NKI development tasks used to train and assess frontier AI models. You’ll assess whether kernels are technically correct, appropriately designed for Trainium hardware, and optimized for the platform.
Your expertise will be particularly valuable when evaluating CUDA-to-NKI migrations, Trainium-specific performance optimization, and cross-platform numerical correctness.
What You’ll Do
- Evaluate NKI kernel development tasks for quality, correctness, completeness, and hardware appropriateness.
- Assess the fidelity and technical quality of CUDA-to-NKI migrations.
- Evaluate Trainium-specific kernel optimization strategies and performance considerations.
- Review tile-based computation approaches and NKI memory-management patterns.
- Assess the appropriate use of SBUF, PSUM, and HBM within kernel implementations.
- Evaluate partition-dimension constraints and DMA orchestration.
- Assess Trainium kernel performance, including: NeuronCore pipeline utilization
- Tensor-engine throughput
- Memory-bandwidth bottlenecks
- Evaluate cross-platform numerical correctness between GPU and Trainium implementations.
- Assess differences arising from accumulation order, rounding behavior, and mixed-precision semantics.
- Provide clear, structured, rubric-based written feedback on kernel tasks and solutions.
What You Bring
- 2+ years of hands-on experience developing or optimizing kernels with Neuron Kernel Interface (NKI) for AWS Trainium or Inferentia2 hardware.
- Strong understanding of NKI development patterns, including: Tile-based computation
- SBUF, PSUM, and HBM memory hierarchy
- Partition-dimension constraints
- DMA orchestration
- Demonstrated experience assessing or performing CUDA-to-NKI kernel migrations.
- Familiarity with Trainium-specific performance profiling and optimization.
- Strong understanding of cross-platform numerical correctness, including: Accumulation order
- Rounding behavior
- Mixed-precision semantics
- Ability to analyze technical implementations with precision and communicate findings clearly in writing.
Preferred Qualifications
Experience in the following areas is highly valuable:
- Hands-on experience with the AWS Neuron SDK.
- Familiarity with Neuron Compiler internals.
- Contributions to NKI kernel libraries or related projects.
- Previous CUDA or Triton kernel development experience.
- Strong understanding of Trainium hardware architecture and capabilities, including: NeuronCore-v2 architecture
- On-chip SRAM topology
- FP32, BF16, FP8, and INT8 data types
- Experience benchmarking machine-learning training workloads on Trn1 or Trn2 instances.
Compensation & Engagement
- Rate: $70–$90/hour
- Annualized Equivalent: $145,600–$187,200
- Location: United States
- Work Arrangement: Fully remote
- Engagement Type: Independent contractor
- Schedule: Flexible
- Payment: Weekly via Stripe or Wise
Annualized compensation is based on 2,080 hours per year for comparison purposes only. Actual earnings depend on the number of hours and projects completed.
Why This Opportunity?
- Apply your Trainium and NKI expertise to advanced AI development.
- Work on technically challenging kernel evaluation and optimization problems.
- Help improve AI systems' understanding of specialized accelerator architectures.
- Evaluate real-world kernel migration, performance, and numerical-correctness challenges.
- Contribute expertise in an increasingly important AI hardware ecosystem.
- Work remotely with a flexible schedule.
- Earn competitive compensation for highly specialized technical expertise.
Contract & Payment Terms
- You will be engaged as an independent contractor.
- Work is fully remote and can be completed on your own schedule.
- Projects may be extended, shortened, or concluded early depending on project needs and performance.
- Your work will not require access to confidential or proprietary information belonging to any employer, client, or institution.
- Payments are made weekly via Stripe or Wise based on services rendered.
- H-1B and STEM OPT candidates cannot be supported at this time.
Equal Opportunity
All qualified applicants will be considered without regard to legally protected characteristics. Reasonable accommodations are available upon request.
Create a free OnJob profile to apply and see your AI match score before you apply. · Posted 27 Aug 2026.
Trainium (NKI) Kernel Expert at Recruitment Room — questions answered
What does the Trainium (NKI) Kernel Expert role at Recruitment Room pay?
The Trainium (NKI) Kernel Expert at Recruitment Room listing states pay of $145,600–$187,200 for the Remote · United States-231 (remote) position. That figure is taken directly from the employer's own posting as published, not estimated or averaged from other roles, and the same listing asks for 2–15 yrs of experience.
Is the Trainium (NKI) Kernel Expert role at Recruitment Room remote?
Yes — Recruitment Room advertises this Trainium (NKI) Kernel Expert role as remote, tied to Remote · United States-231, and it is listed as contract work. Remote terms come from the employer's own posting, so confirm the expected working hours, timezone and any on-site days with Recruitment Room before you apply.
What skills does the Trainium (NKI) Kernel Expert role at Recruitment Room require?
The Trainium (NKI) Kernel Expert at Recruitment Room listing names CUDA, AWS and SDK. Those are the skills the employer put on the posting itself, so they are the ones worth matching in your profile and covering first in an interview.
How much experience do you need for the Trainium (NKI) Kernel Expert role at Recruitment Room?
Recruitment Room asks for 2–15 yrs of experience on this Trainium (NKI) Kernel Expert posting, alongside CUDA, AWS and SDK. Employers commonly consider candidates slightly under a stated band when the listed skills line up.
Is the Trainium (NKI) Kernel Expert role at Recruitment Room still open?
The Trainium (NKI) Kernel Expert posting at Recruitment Room was open at OnJob's last check of the employer's careers page, having been published on 27 August 2026. OnJob re-checks source listings on each build and marks a role closed once it disappears, but listings can close without notice, so the employer's own page is the final word.
How do you apply for the Trainium (NKI) Kernel Expert role at Recruitment Room?
Apply to the Trainium (NKI) Kernel Expert at Recruitment Room role through OnJob with a free profile: OnJob scores your fit against the listing, shows the skills lowering that score, and submits an ATS-ready profile. Creating a profile is free and needs no card.
Explore more on OnJob
Hiring for a role like this?
Post a job on OnJob and reach AI-matched candidates.
