Trainium (NKI) Kernel Expert
$70 - $90 per hourTemporary
Mercor
Evaluate the quality, correctness, and hardware-appropriateness of Neuron Kernel Interface (NKI) development tasks used to train and evaluate a frontier AI lab's models. You'll assess CUDA→NKI migration fidelity, Trainium-specific performance-optimization quality, and cross-platform numerical-correctness standards, and provide clear, rubric-based written feedback.
Basic Qualifications
- 2+ years of hands-on experience developing or optimizing kernels using the Neuron Kernel Interface (NKI) targeting AWS Trainium/Inferentia2 hardware
- Strong understanding of NKI-specific development patterns: tile-based computation, SBUF/PSUM/HBM memory-hierarchy management, partition-dimension constraints, and DMA orchestration
- Demonstrated experience assessing CUDA→NKI migration quality
- Familiarity with Trainium-specific performance profiling (NeuronCore pipeline utilization, tensor-engine throughput, memory-bandwidth bottlenecks)
- Experience defining or evaluating cross-platform numerical-correctness standards (GPU vs Trainium accumulation order, rounding behavior, mixed-precision semantics)
Preferred Qualifications
- Direct 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 (NeuronCore-v2 architecture, on-chip SRAM topology, supported data types: FP32/BF16/FP8/INT8)
- Experience benchmarking ML training workloads on Trn1/Trn2 instances
Vacancy posted a month ago
Similar jobs that could be interesting for youBased on the Trainium (NKI) Kernel Expert in San Francisco, CA vacancy
$70 - $90 per hour
...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...SuggestedContract workSummer workRemote work$70 - $90 per hour
...Evaluate the quality, correctness, and completeness of GPU/accelerator kernel development tasks used to train and evaluate a frontier AI lab's... ...GPU/accelerator kernels in at least two of: CUDA, Triton, NKI, or Pallas (JAX) • Strong understanding of numerical-correctness...SuggestedContract work$60k - $74k
...sleep well so they live well is at the core of what we do. The right mattress matters, and so does the right career. Our Sleep Experts® are at the heart of what we do: helping customers find their dream mattress and get better sleep. By joining our team, you'll get...SuggestedLocal areaMattress Firm
San Francisco, CA2 days ago- ...communities. The Role We are looking for an experienced pharmacovigilance professional to serve as our Senior Safety Domain Expert for Veeva Falcon. In this role, you will bridge hands-on case processing execution with high-level strategic vision for the future...SuggestedWork at officeLocal areaWork from homeShift work
Veeva Systems
San Francisco, CA9 hours ago - ...PHP Python React Rust TypeScript ASP.NET NodeJS .NET Framework Golang HTML5 More: Job Title: Coding Expert Job Type: Contract (Part-time) Location: Remote (US, Canada, United Kingdom, Ireland, Australia, New Zealand) Job Summary:...SuggestedContract workPart timeRemote work
SwissDevJobs.ch
San Francisco, CA16 hours ago $60k - $74k
...sleep well so they live well is at the core of what we do. The right mattress matters, and so does the right career. Our Sleep Experts are at the heart of what we do: helping customers find their dream mattress and get better sleep. By joining our team, you'll get...Local areaMattress Firm
Colma, CA1 day ago$50 per hour
...AI projects with leading LLM companies. Pay rate: $50+/hour (depends on role and candidate expertise) Assessment: Shortlisted experts complete an evaluation before selection. Assignments: Contract roles with defined start/end dates; up to 40 hrs/week. About...Contract workRemote workFlexible hoursTuring
San Francisco, CA4 days ago- ...connection, and community. Join us and be part of a team that's changing lives every day! JOB SUMMARY The Sales and Service Expert (SSE) delivers an outstanding and well-rounded sales and service experience to all guests and members through a compelling approach...Work experience placementWork at officeLocal area
24 Hour Fitness
San Francisco, CA1 day ago $50 per hour
...projects with leading LLM companies. Pay rate: 50+/hour (depends on role and candidate expertise). Assessment: Shortlisted experts complete an evaluation before selection. Assignments: Contract roles with defined start/end dates; up to 40 hrs/week. About...Contract workRemote workFlexible hoursTuring Inc
San Francisco, CA4 days ago$120 per hour
About the role We are hiring expert Evaluators in Customer success / support operations to review and assess AI-generated work products (documents, spreadsheets, and slide decks) for accuracy, rigor, and domain quality. You will apply deep subject‑matter expertise to grade...Remote jobHourly payWork at office$110 per hour
We are seeking U.S.-based experts with direct underwriting experience in the Excess & Specialty Insurance market, particularly underwriting SME/commercial accounts . Engagement Details Initial pilot: 4 experts Duration: 1 week Commitment: ~20 hours per expert Potential...Remote job- ...customer relationships and keep the customer at the center Provide technical and product knowledge information as a subject matter expert Drive individual sales performance and KPIs while supporting company initiatives Maintain customer relationships during and after...
$60 per hour
Prolific is seeking Accountant Subject Matter Experts to join their Expert Network to provide professional insights in AI training. Responsibilities include evaluating financial documents, providing AI model guidance, and ensuring data integrity. The ideal candidates will...Remote jobHourly payWork from homeFlexible hours$60 per hour
Prolific seeks Chemistry Experts and Chemical Engineers to join our Expert Network and evaluate AI models using your chemical expertise. Successful candidates will be invited to assess and review AI-generated chemistry tasks and ensure their accuracy. Compensation can reach...Remote jobHourly payFlexible hours- ...life sciences industry, committed to making a positive impact on its customers, employees, and communities. The Role As a Product Expert, you will be the ultimate bridge between our software development teams, internal business stakeholders, and our customers. You will...Work at officeWork from home
- ...end users of Maps or test the location results of user queries submitted via Web Search or Maps interface. The Mapping City/Country Expert will be responsible for high‑impact work in improving city transit data and providing country‑specific expertise. The City/Country...Local area
$17 - $25 per hour
...investors include Benchmark , General Catalyst , Peter Thiel , Adam D'Angelo , Larry Summers , and Jack Dorsey . Position: AI Safety Experts — English & Malay Type: Contract Compensation: $17-$25/hour Location: Remote Role Responsibilities Red team conversational AI...Contract workSummer workRemote work$12 per hour
...General Catalyst , Peter Thiel , Adam D'Angelo , Larry Summers , and Jack Dorsey . Position: PDF Annotation & Transcription Experts – Telugu Type: Contract Compensation: $13/hour Location: Remote Role Responsibilities Source documents by...Remote jobContract workSummer workMercor
San Francisco, CA6 days ago$75 - $100 per hour
...Join to apply for the Physics PhD - Expert Trainer role at Handshake Continue with Google Continue with Google 3 days ago Be among the first 25 applicants Join to apply for the Physics PhD - Expert Trainer role at Handshake This range is provided by Handshake...Contract workFreelanceInternshipH1bRemote workVisa sponsorship10 hours per weekFlexible hoursHandshake
San Francisco, CA4 days ago$90k - $200k
...We are looking for a Vice President to act as the lead assistant on small/medium projects for our Delay and/or Quantum experts involved in domestic and international arbitration and litigation. As a Vice President, you’ll work with projects in infrastructure, energy, industrial...Kroll
San Francisco, CA4 days ago- ...Job Description Job Description Developer & Infrastructure Expert Role Type: Contractor Location: Remote Job Overview We are seeking experienced Developer & Infrastructure Experts to evaluate AI-powered workflows across software development, cloud infrastructure...Remote jobFor contractors
YO AI Labs
San Francisco, CA3 days ago $70.72k
...plan, 401(k), and access to free, year-round money coaches. That’s how we’re UNSTOPPABLE for our employees!Job Overview Sr Mobile Expert are our most capable experts with a passion for delivering best-in-class customer service experience while also leveraging their...Hourly payFull timeTemporary workPart timeWork experience placementLocal areaFlexible hours$500 per month
...Summers , and Jack Dorsey . Position: CUDA Engineering Expert Type: Contract Compensation: $300 per task Location... ...Remote Role Responsibilities Analyze and optimize GPU kernels for performance, efficiency, and hardware utilization. Use...Remote jobContract workSummer work- ...Job Description Job Description GPU Programming Expert - Remote Job Type: Contractor Location: Remote Job Overview We... ...software using CUDA, WebGPU, or GLSL . Profile and optimize GPU kernels and shaders for performance and efficiency. Develop host-...Remote jobFor contractors
- A leading IT systems integrator is seeking a Senior Continuous Integration Engineer in the Bay Area, California. This role focuses on integrating two medical device systems to support radiology testing. The ideal candidate has experience with continuous integration, Jenkins...
$120 per hour
Prolific, located in San Francisco, is looking for Licensed Pharmacists to train and evaluate AI models. You will guide these systems by reviewing AI-generated pharmaceutical scenarios and providing your expertise. With competitive pay rates of up to $120 USD per hour based...Remote jobHourly payWork from homeFlexible hours- ...Job Description Job Description Job Title: Robotics & Mechatronics AI Expert Role Type: Contractor Location: Remote Job Overview We are seeking experienced Robotics and Mechatronics Experts to contribute their technical expertise to a project focused on...Remote jobFor contractors
YO AI Labs
San Francisco, CA3 days ago - The Federal Bureau of Investigation (FBI) is seeking a special agent with a psychology or counseling background to help protect national security. You will plan and conduct investigations across cyber, fraud, and public safety domains, applying analytical thinking, data...Work at office
- Nurix Therapeutics is seeking a Scientist I, Bioanalytical to lead bioanalytical strategies for DACs, spanning LC-MS/MS and hybrid assays to analyze complex bioconjugates in vitro and in vivo. The role interfaces with DMPK, Proteomics, Pharmacology, and Chemistry to inform...
- Nurix Therapeutics is seeking a Scientist to lead bioanalytical strategies for DACs, focusing on LC-MS/MS and hybrid approaches to quantify and characterize complex bioconjugates in vitro and in vivo. You will collaborate across DMPK, Proteomics, Pharmacology, and Chemistry...
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Trainium (NKI) Kernel Expert. Be the first to apply!
