Cupertino-Based Software Development Manager, NKI & Trainium

Amazon Web Services (AWS); Cupertino (CA); On-site; USD 213,000 - 288,000; Full time

Benefits offered by this job

Sign-on payments

RSUs

Job summary

Amazon Annapurna Labs in Cupertino, CA is seeking a Software Development Manager (SDM III) to lead the NKI compiler team across Trainium hardware. You will mentor engineers, define the roadmap, and drive execution to deliver high-performance compiler features.

You will collaborate with frameworks, kernel development, runtime, and pre-silicon design to bring new features to market, balancing speed and quality while focusing on customer impact and ML workloads.

Qualifications

• 2+ years of engineering team management experience.
• 6+ years working directly within engineering teams.
• 4+ years designing or architecting (design patterns, reliability and scaling) of new and existing systems.
• Experience partnering with product or program management teams.
• Understanding of compilers (resource management, instruction scheduling, code generation, and compute graph optimization)
• Strong software design fundamentals and excellent system-level coding skills.

Responsibilities

• Lead, grow, and mentor a team of compiler engineers, including hiring, career development, and team health
• Own execution and delivery of NKI compiler features across release cycles, balancing scope, quality, and timelines
• Set technical direction and roadmap for your area of the NKI compiler in partnership with senior engineers
• Partner across teams, including frameworks, kernel development, runtime, hardware/pre-silicon design, and product management, to bring new features to market
• Drive resolution of complex, high-priority technical issues, staying close enough to the code and hardware to guide the team
• Represent your team's work and priorities to senior leadership and cross-organizational stakeholders
• Anchor decisions in customer impact, ensuring the team builds what unblocks real ML workloads on Trainium

Skills

Engineering team management

Engineering team experience

System design

Product/PM collaboration

Compiler concepts

Software design fundamentals

Education

MS/PhD in CS

Job description

Amazon Annapurna Labs in Cupertino, CA is seeking a Software Development Manager (SDM III) to lead the NKI compiler team across Trainium hardware. You will mentor engineers, define the roadmap, and drive execution to deliver high-performance compiler features.

You will collaborate with frameworks, kernel development, runtime, and pre-silicon design to bring new features to market, balancing speed and quality while focusing on customer impact and ML workloads.