Software Development Manager: NKI Compiler for Neuron

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

Benefits offered by this job

Health insurance

401(k) matching

Paid time off

Parental leave

Job summary

Amazon Annapurna Labs is seeking a Software Development Manager (SDM III) for the NKI Compiler on AWS Trainium. You will lead a team of experienced compiler engineers, drive delivery across release cycles, and set technical direction.

Collaborating with frameworks, kernel development, runtime, hardware design, and product management, you will translate customer needs into concrete technical investments and mentor engineers to grow into senior contributors.

Qualifications

• 2+ years of engineering team management experience.
• 6+ years working directly within engineering teams.
• 4+ years designing or architecting scalable systems.
• Experience partnering with product or program management.
• Understanding of compilers, resource management, and optimization.
• Strong software design fundamentals and system-level coding.

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 leadership

People management

Compiler knowledge

Strategic planning

Education

Bachelor's degree in CS or related field

MS/PhD preferred

Job description

Amazon Annapurna Labs is seeking a Software Development Manager (SDM III) for the NKI Compiler on AWS Trainium. You will lead a team of experienced compiler engineers, drive delivery across release cycles, and set technical direction.

Collaborating with frameworks, kernel development, runtime, hardware design, and product management, you will translate customer needs into concrete technical investments and mentor engineers to grow into senior contributors.