Taro Logo

Sr. Machine Learning Compiler Engineer, Annapurna Labs

AWS Machine Learning accelerator division developing custom silicon chips for ML inference and training
Machine Learning
Senior Software Engineer
In-Person
5,000+ Employees
5+ years of experience
AI · Enterprise SaaS

Description For Sr. Machine Learning Compiler Engineer, Annapurna Labs

AWS Machine Learning accelerators are revolutionizing cloud computing with custom silicon chips - Inferentia for ML inference and Trainium for ML training. These are powered by the AWS Neuron SDK, featuring a deep learning compiler and runtime with native integration for popular ML frameworks. The Tel Aviv team will be crucial in developing the Neuron compiler stack, translating neural network models into high-performance execution on AWS custom hardware.

As a Senior Machine Learning Compiler Engineer, you'll join Amazon Annapurna Labs' AWS Neuron team, working across software, hardware, and silicon engineering. You'll lead compiler technology development, optimize ML workloads, and contribute to hardware development from pre-silicon design through production deployment. The role offers opportunities to influence architecture, mentor engineers, and collaborate with AWS service teams.

The position requires strong technical leadership, with responsibilities including architecting critical features, publishing research, and mentoring experienced engineers. You'll work closely with AWS ML services teams and contribute to various stages of product development. While ML and AI accelerator experience is preferred, the role primarily seeks innovative engineers passionate about scaling complex systems.

This is a founding team position in Tel Aviv, offering the unique opportunity to shape both technical direction and team culture while contributing to AWS's global ML infrastructure. The work directly impacts major customers like Snap, Autodesk, and Amazon's own services, making it an excellent opportunity for engineers looking to make a significant impact in the ML/AI infrastructure space.

Last updated 17 days ago

Responsibilities For Sr. Machine Learning Compiler Engineer, Annapurna Labs

  • Lead development of cutting-edge compiler technologies
  • Optimize large-scale ML workloads
  • Influence architecture and mentor other engineers
  • Collaborate with AWS services teams
  • Work on projects spanning full product lifecycle from pre-silicon design to production
  • Implement business-critical features
  • Publish cutting-edge research

Requirements For Sr. Machine Learning Compiler Engineer, Annapurna Labs

Python
  • 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 of new and existing systems experience
  • 5+ years of full software development life cycle experience
  • Experience as a mentor, tech lead or leading an engineering team
  • Bachelor's degree in computer science or equivalent

Benefits For Sr. Machine Learning Compiler Engineer, Annapurna Labs

Medical Insurance
Vision Insurance
Dental Insurance
  • Work-life harmony
  • Flexible working culture
  • Career development opportunities
  • Mentorship programs
  • Knowledge-sharing resources
  • Inclusive team culture
  • Employee-led affinity groups

Jobs Related To Annapurna Labs Ltd. Sr. Machine Learning Compiler Engineer, Annapurna Labs