Sr. Compiler Engineer III - Machine Learning, Annapurna Labs

$151,300 - $261,500/Yr

Amazon - Seattle, WA

Full-time - Mid Level

Remote - Seattle, WA

5,001-10,000 employees

Sporting Goods, Hobby, Musical Instrument, Book, and Miscellaneous Retailers

About the position

The Sr. Compiler Engineer III position at Annapurna Labs focuses on developing and scaling a compiler for AWS Machine Learning accelerators, specifically designed to handle large ML workloads efficiently. The role involves architecting and implementing critical features, collaborating with various teams, and contributing to cutting-edge research in the field of machine learning and compiler technology.

Responsibilities

• Solve challenging technical problems across the software stack.
• Design, implement, test, deploy, and maintain innovative software solutions.
• Build high-quality, highly available products.
• Research implementations to enhance customer experiences.
• Create metrics, implement automation, and resolve software defects.
• Participate in design discussions and code reviews.
• Work cross-functionally to influence business decisions with technical input.

Requirements

• 6+ years of experience in leading design or architecture of systems.
• 5+ years of experience in developing compiler features and optimizations.
• Proficiency in C++, C, or Python.

Nice-to-haves

• Masters degree or PhD in computer science or equivalent.
• Experience optimizing TensorFlow, PyTorch, or JAX deep learning models.
• Experience with toolchains like LLVM, OpenXLA/XLA, MLIR, TVM.

Benefits

• Flexible working hours and hybrid work model.
• Mentorship and career growth opportunities.
• Inclusive team culture and employee-led affinity groups.
• Work-life balance initiatives.