Taro Logo

ML Compiler Engineer II - Automated Reasoning Science, Annapurna Labs

Amazon is a global technology company and the world's largest e-commerce retailer.
$129,300 - $223,600
Machine Learning
Mid-Level Software Engineer
In-Person
5,000+ Employees
3+ years of experience
AI · Enterprise SaaS
This job posting may no longer be active. You may be interested in these related jobs instead:

Description For ML Compiler Engineer II - Automated Reasoning Science, Annapurna Labs

The AWS Neuron Compiler team is seeking a skilled ML Compiler Engineer II to join their cutting-edge development efforts. This role focuses on building a state-of-the-art deep learning compiler stack that optimizes application models across various domains, including Large Language Models and Vision, working with frameworks like PyTorch, TensorFlow, and JAX. The position is part of the Automated Reasoning Group, where you'll be developing advanced tooling for the Neuron Compiler's accuracy and reliability.

The role requires expertise in C++/Python programming and compiler development, with a focus on LLVM-based systems. You'll be working with AWS's custom-built Machine Learning accelerators (Inferentia/Trainium) that power innovative solutions like Generative AI. The position offers a competitive salary range of $129,300 to $223,600, depending on location and experience, plus comprehensive benefits including equity and sign-on bonuses.

As part of Amazon's AWS division, you'll be working at the forefront of machine learning infrastructure, helping to build and optimize the tools that power next-generation AI applications. The role requires a strong technical background, with preferred qualifications including a Masters or PhD in computer science and experience with deep learning frameworks. You'll be joining a science-focused team that pushes the boundaries of what's possible in compiler technology and machine learning optimization.

The position offers the opportunity to work in major tech hubs including Seattle, Cupertino, Austin, or Toronto, with a collaborative team environment and access to cutting-edge technology. This is an excellent opportunity for someone passionate about compiler development, machine learning, and high-performance computing to make a significant impact on AWS's machine learning infrastructure.

Last updated 2 months ago

Responsibilities For ML Compiler Engineer II - Automated Reasoning Science, Annapurna Labs

  • Develop and maintain tooling for Neuron Compiler's accuracy and reliability
  • Lead efforts building fuzzers and specification synthesis tooling for LLVM-based compiler
  • Work in a science-focused team to push technological boundaries
  • Optimize application models across diverse domains including Large Language and Vision

Requirements For ML Compiler Engineer II - Automated Reasoning Science, Annapurna Labs

Python
  • 3+ years of leading design or architecture experience
  • 2+ years of experience in developing compiler features and optimizations
  • Proficiency in C++ and Python programming
  • Familiarity with LLVM, including knowledge of abstract interpretation and polyhedral domains
  • Demonstrated scientific approach to software engineering problems

Benefits For ML Compiler Engineer II - Automated Reasoning Science, Annapurna Labs

Medical Insurance
401k
  • Medical, financial, and other benefits
  • Equity compensation
  • Sign-on payments
  • Competitive base pay based on location

Interested in this job?