Senior ML Compiler Engineer (NKI) – Scale ML Workloads

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

Job summary

Amazon Annapurna Labs (U.S.) Inc. seeks a Sr. Machine Learning Compiler Engineer to lead the ground-up development and scaling of a compiler for the world's largest ML workloads. You will architect and implement core features, publish cutting-edge work, and mentor a team of engineers, while collaborating closely with AWS ML teams.

A background in machine learning and AI accelerators is preferred but not required. Candidates must be willing to relocate to Seattle, Cupertino, or Austin.

Qualifications

• 5+ years of non-internship professional software development experience.
• 5+ years of programming with at least one software programming language experience.
• 5+ years of leading design or architecture (design patterns, reliability and scaling) of new and existing systems experience.
• 5+ years of full software development life cycle, including coding standards, code reviews, source control management, build processes, testing, and operations experience.

Responsibilities

• Develop and scale a compiler for large ML workloads.
• Architect and implement business-critical features and mentor engineers.
• Publish cutting-edge work and collaborate with AWS ML teams.
• Leverage technical communication skills as hands-on partner to teams.

Skills

Software development

Programming experience

System design/architecture

SDLC / DevOps

Education

Bachelor's degree in Computer Science or equivalent

Job description

Amazon Annapurna Labs (U.S.) Inc. seeks a Sr. Machine Learning Compiler Engineer to lead the ground-up development and scaling of a compiler for the world's largest ML workloads. You will architect and implement core features, publish cutting-edge work, and mentor a team of engineers, while collaborating closely with AWS ML teams.

A background in machine learning and AI accelerators is preferred but not required. Candidates must be willing to relocate to Seattle, Cupertino, or Austin.