ML Kernel Performance Engineer

Home > Other programming jobs

ML Kernel Performance Engineer in USA

Amazon
Full time
Email

USA
09/03/2026
Report

Cupertino, CA

Responsibilities

Design and implement high-performance compute kernels for machine-learning operations using the Neuron architecture and programming models.
Analyze and optimize kernel-level performance across multiple generations of Neuron hardware.
Use profiling tools to identify and resolve performance bottlenecks.
Implement compiler optimizations including fusion, sharding, tiling, and scheduling.
Work directly with customers to enable and optimize their machine-learning models on AWS accelerators.
Collaborate with compiler, runtime, framework, and hardware teams on kernel optimization techniques.
Create metrics, implement automation, resolve software defects, participate in design discussions and code reviews, and provide technical input to business decisions.
Publish research and mentor experienced engineers.

Requirements

At least 3 years of non-internship professional software development experience.
At least 2 years of non-internship experience designing or architecting new and existing systems, including design patterns, reliability, and scaling.
Experience programming with at least one software programming language.
Preferred: at least 3 years of full software development life-cycle experience, including coding standards, code reviews, source control management, build processes, testing, and operations.
Preferred: bachelor’s degree in computer science or equivalent.

Benefits

Comprehensive health insurance covering medical, dental, vision, prescription, Basic Life and AD&D insurance, and optional supplemental life plans.
Employee Assistance Program, mental health support, medical advice line, flexible spending accounts, and adoption and surrogacy reimbursement coverage.
401(k) matching, paid time off, and parental leave.
Sign-on payments and restricted stock units are included in the overall Amazon package.
Flexible working hours and an emphasis on work-life balance.
Inclusive culture with employee-led affinity groups and ongoing learning experiences.

Salary: $165k - $224k/yr

Amazon

Amazon is a global leader in e-commerce and cloud computing, connecting millions of customers with a vast selection of products and services. The company thrives on innovation, customer obsession, and a relentless focus on convenience and reliability. With a culture that empowers teams to experim...

Similar positions

Senior Test Automation Engineer

City of New York
Full time

USA
09/13/2026

Brooklyn, NY

Backend Engineer, Security Platform

Vannevar
Full time

USA
09/13/2026
Salary: $150k - $215k/yr

Remote

Agentic Security Engineer

NXP Semiconductors
Full time

USA
09/13/2026

Austin, TX

Sr Application Security Engineer

Henry Schein One
Full time

USA
09/13/2026
Salary: $125k-$160k

Remote

Senior DevOps Engineer, Infrastructure & Reliabili

Worth AI
Full time

USA
09/13/2026

Remote