Senior Applied Scientist, Annapurna ML

About the role

We are seeking a Senior Applied Scientist to work at the intersection of Artificial Intelligence and program analysis, raising the code quality bar in our state-of-the-art deep learning compiler stack. This stack optimizes application models across diverse domains, including Large Language and Vision, from leading frameworks such as PyTorch, TensorFlow, and JAX.

You will collaborate closely with our custom-built Machine Learning accelerators, Inferentia and Trainium, which power advanced ML capabilities and underpin Generative AI. Your work will involve designing, developing, and deploying analyzers for ML compiler stages and compiler IRs, architecting business-critical tooling, publishing research, and mentoring a team of experienced scientists and engineers.

Responsibilities

• Design, develop, and deploy analyzers for ML compiler stages and compiler IRs.
• Architect and implement business-critical tooling for our deep learning compiler stack.
• Publish research and contribute to leading-edge design in compiler technology and deep-learning systems software.
• Mentor a team of experienced scientists and engineers.
• Tackle crucial challenges in program analysis, compilers, and AI accelerators alongside a talented engineering team.

Requirements

• Strong experience in programming languages, compilers, program analyzers, or program synthesis engines.
• Technical credibility and curiosity as a trusted scientist, capable of innovating on behalf of customers.
• Background in machine learning and AI accelerators is preferred but not required.

Benefits

• Employee-led affinity groups fostering an inclusive culture.
• Ongoing learning experiences, including Conversations on Race and Ethnicity (CORE) and AmazeCon (diversity) conferences.
• Mentorship and career growth resources to develop into a better-rounded professional.
• Work-life harmony with flexibility as part of our working culture.