NKI Kernel Optimization Specialist
$70 - $90 per hourMercor
Job Description
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:
- For any help or support, reach out to: View email address on ziprecruiter.com
PS: Our team reviews applications daily. Please complete your AI interview and application steps to be considered for this opportunity.
Vacancy posted 5 days ago
Similar jobs that could be interesting for youBased on the NKI Kernel Optimization Specialist in San Francisco, CA vacancy
- ...quality, correctness, and hardware-appropriateness of Neuron Kernel Interface (NKI) development tasks used to train and evaluate a frontier... ...→NKI migration fidelity, Trainium-specific performance-optimization quality, and cross-platform numerical-correctness standards...Suggested
- A leading tech company in San Francisco is seeking an ASO Specialist to enhance app visibility and optimize strategies for user acquisition. This role demands a detail-oriented individual with 3+ years of experience in ASO or mobile growth marketing. You will conduct keyword...Suggested
$80 per hour
...Research, Industrial Engineering, Applied Mathematics, or a closely related field. 2-6 years of professional or research experience in optimization, logistics, or decision science. Strong command of mathematical programming, probabilistic modeling, and algorithmic methods....SuggestedContract workSummer workRemote work$150 per hour
...Search Systems Type: Contract Compensation: $80–$150/hour Location: Remote Role Responsibilities Develop and optimize agentic search systems for production environments. Own relevance and retrieval processes to ensure systems meet user needs....SuggestedRemote jobContract workSummer work$75.6k - $194.8k
Associate or Full Specialist - Liu labThe Specialist will execute a research program at the intersection of RNA-based nanomedicine, CRISPR... .... Specific duties include:Design, synthesis, and biological optimization of lipid nanoparticle (LNP) formulations for intracranial...SuggestedLocal area- ...spend decisions with clear data points, and driving process improvements across a fast-paced contracting environment. You will collaborate with global teams across research, procurement, shared services, and finance to optimize workflows and #J-18808-Ljbffr GenentechContract work
$150.1k - $227k
...future of AI, and you are the future of Salesforce.Lead System Specialist - Email Platform & Consent InfrastructureThe Messaging... ...Salesforce Data Cloud. We prioritize system stability, process optimization, and execution excellence to ensure high-stakes marketing communications...Full time- A leading AI research firm is seeking an Operations Research Professional to craft precise operations research prompts for AI assessments. This remote position requires a master's degree or higher in Operations Research or a related field, along with 2-6 years of relevant...Remote job
- DoorDash is seeking a highly analytical, execution-focused Cost Optimization Analyst to drive cost improvement initiatives within the Commerce Platform. The role partners with Finance and works with Product, Engineering, Analytics, and Operations to identify and execute...
$55k - $194.8k
Manglik and Arkin labs - Junior/Assistant/Associate/Full Specialist Department of Pharmaceutical Chemistry UCSF The labs of Aashish Manglik... ...will take increasing ownership of a project area, help optimize workflows, and contribute to troubleshooting, data interpretation...Local area$95k - $130k
Sr. Analyst, Service Experience Optimization Full-time SquareTrade, Inc. d/b/a Allstate Protection Plans ("SquareTrade") is one of the fastest-growing companies in its industry. We’re transforming a $30B market through service innovation and an unwavering focus on customer...Full timeContract workWork at officeWork visaFlexible hours- ...workforce management, staffing office operations, or clinical labor consultingDeep interest or experience in workforce-composition optimization and labor assessment. Familiar with strategies for reducing clinician burnout, improving retention, and modernizing care...ApprenticeshipWork at officeEasy work
$132.89k - $166.11k
...ownership, this is your team. What you’ll do As a Banking Specialist on our newly formed Banking Team, you will have the... ...Finance, and Controllers to continue generating net-new revenue by optimizing their adoption of our business account suite of products. As a...Full timeWork at officeRemote workWork from home$55 - $62 per hour
DescriptionKforce's client is seeking a Sourcing Specialist for 2 remote roles.Summary:In this role, you'd be joining Clients Strategic... ...based on industry trends and internal spend data to generate optimization opportunities* Supporting sourcing plans and supplier spend...Contract workRemote work$207k - $275k
...workloads, and those new services need a market-maker. As a Solution Specialist for AI Runtime Services, you open new market opportunities... ...working with AI runtime systems (model serving, inference optimization, containerized workload execution, or real-time ML pipelines)...Permanent employmentFull timeTemporary workCasual workWork at officeFlexible hours$179k - $242.2k
...sourcing new business- 5+ years of customer-facing product development and product management, including defining, launching, and optimizing product and services experience- 5+ years of Go-To-Market, Business Development, Sales, or Consulting experience- 5+ years of...Local areaWorldwideFlexible hours$110k - $130k
...profitability. Success in this role means translating complex technical findings into clear business recommendations that help clients optimize their critical infrastructure investments.Your day-to-day tasks will include:Conducting comprehensive Critical Environment Gap...Full timeWork at officeLocal areaRemote work- Overview Job Title: Lead Generation Specialist (Remote) Job Timings: 8 am - 1 pm Pacific Standard Time (Part-time) Industry Focus: B2B... ...clinical teams with innovative solutions. Our mission is to make optimal care accessible, providing healthcare professionals with the...Hourly payPart timeRemote workFlexible hours
$57k - $201.7k
...translation to humans. Responsibilities include: Development and optimization of electroplating approaches for 512-channel flexible probes... ...recording technologies.Required Qualifications: Junior Specialist - Baccalaureate degree (or equivalent degree) or at least four...Flexible hours$125k - $135k
...that currently we are looking for Health Information Technology Specialists to support an IQVIA pharmaceutical client partnershipThis... ...and other cross functional partners. TShare insights around optimizing EHR systems to address gaps in patient care including Inpatient...Full timePart timeImmediate startWorldwideNight shift$93.2k - $194.8k
Full Specialist Microbiome of Rheumatic DiseasesThe Nayak Lab (nayak.ucsf.edu) is seeking a Full Specialist with expertise in microbiome... ..., and computational analysis. • Develop, implement, optimize, and troubleshoot new assays, making innovative contributions...Traineeship- ...and families as they navigate through complex healthcare environments. Primary goals are to increase patient satisfaction, and to optimize care and outcomes. Acts as a communication liaison to understand the patient's non-clinical individual needs, desires, and concerns...
- ...supplies, pharmacy, purchased services), clinical operations (reducing length of stay, increasing throughput) and workforce management (optimizing composition of clinical and non-clinical labor); open to backgrounds working for GPOs, pharmacy companies, medical device...ApprenticeshipEasy work
- ...experience with at least one of the following areas: Payment Integrity including Fraud, Waste & Abuse; Medical & Pharmacy Cost Optimization; Network optimization; Administration & Operational Efficiency; Care Management; Risk AdjustmentUnderstanding of the healthcare...ApprenticeshipWorldwideEasy work
- Xcel Agency Inc. is seeking a remote Email Marketing Specialist to manage and optimize email marketing campaigns. Responsibilities include designing HTML emails, managing A/B tests, and creating landing pages. The role demands strong technical skills and an understanding...Remote job
- Delta Dental of California is seeking an Epic Patient Access & Experience Applications Lead to own the design, implementation, and optimization of Epic patient access applications. The role balances hands-on Epic analyst work with people leadership, guiding a high-...
- ...Manager in San Francisco. This role involves executing ad campaigns, analyzing performance, and collaborating with sales teams to optimize results. Candidates should have at least 2 years of experience in online advertising and a relevant degree. Strong communication and...
- ...and Regional Marketing to deliver high-performing, locally relevant experiences. What Will You Do? As a Manager, Landing Page Optimization at Airwallex, you will lead landing page optimization and conversion rate improvement across the U.S. and EMEA acquisition funnels...Worldwide
- Gap Inc. is transforming how its brands use data and AI to shape pricing decisions. The Manager, Pricing Optimization will run models, build scenarios, and distill insights for Old Navy’s pricing and promotions. You report to the Sr. Manager, Pricing Optimization and own...
- ...multinational advertising campaigns, formulate localized advertising strategies. Analyze data and performance of campaigns, detailed optimization to reaching client’s KPI. Co-work with sales managers, account managers, product managers and on various optimization related...Odd jobWork experience placementShift workWeekend work
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to NKI Kernel Optimization Specialist. Be the first to apply!
Related searches
- cost specialist San Francisco, CA
- strategic sourcing specialist San Francisco, CA
- absence management specialist San Francisco, CA
- authorization specialist San Francisco, CA
- treasury specialist San Francisco, CA
- print production specialist San Francisco, CA
- workforce management specialist San Francisco, CA
- wellness specialist San Francisco, CA
- helpdesk specialist San Francisco, CA
- program eligibility specialist San Francisco, CA