Principal Applied Scientist, Neuron ARG

The AWS Neuron Compiler team is seeking applied scientists to develop a state-of-the-art deep learning compiler stack. This role involves optimizing application models across diverse domains, including Large Language and Vision, from frameworks like PyTorch, TensorFlow, and JAX. You'll work with custom-built Machine Learning accelerators like Inferentia/Trainium, powering advanced ML capabilities and Generative AI solutions.

As an Applied Scientist in the Automated Reasoning Group, you'll design, develop, and deploy analyzers for ML compiler stages and IRs. You'll contribute to DSL design and semantics for neural architectures, architect critical tooling, publish research, and mentor a team of scientists and engineers.

Key responsibilities include:

• Tackling crucial challenges in compiler technology and deep-learning systems software
• Contributing to cutting-edge design and research
• Designing and implementing analyzers for compiler stages and IRs
• Architecting and deploying business-critical tooling
• Publishing research in the field
• Mentoring a team of experienced scientists and engineers

Required qualifications:

• PhD in computer science or related field
• Strong publication record
• Hands-on experience in compiler or tool development

Preferred qualifications:

• Publications in programming languages, compilers, static analysis, or verification
• Experience mentoring junior researchers
• Track record of working with compiler users and developers

The role offers flexibility with options to work in the office daily or in a hybrid model near U.S. Amazon offices. AWS values diverse experiences and encourages applications from candidates with non-traditional backgrounds.

Amazon is committed to diversity, inclusion, and equal opportunity employment.