ML Compiler Engineer, AWS Neuron, Annapurna Labs

Amazon Web Services (AWS) is the world's most comprehensive and broadly adopted cloud platform, pioneering cloud computing and continuously innovating.
Machine Learning
Mid-Level Software Engineer
Contact Company
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:
System Development Engineer, AGI - Modeling Services

System Development Engineer role focusing on AGI and LLM infrastructure development and maintenance at Amazon, requiring 3+ years of experience in systems engineering.

Machine Learning Engineer, Workforce Solutions - Analytics and Tech

Machine Learning Engineer role at Amazon focusing on AI/ML systems and LLMs to solve workforce challenges, offering competitive compensation and cutting-edge tech exposure.

ASIC Design Engineer, Cloud-Scale Machine Learning Acceleration team

ASIC Design Engineer role at AWS, focusing on machine learning hardware optimization and development of high-performance RTL designs for cloud infrastructure.

Machine Learning Engineer, Robotic Storage Technologies - Simulation & Machine Learning

Machine Learning Engineer role at Amazon Robotics, focusing on AI-driven warehouse optimization and robotic storage solutions.

Software Dev Engineer II, Amazon Q

Software Development Engineer II position at Amazon Q, focusing on AI-powered developer tools with competitive compensation and comprehensive benefits.

Description For ML Compiler Engineer, AWS Neuron, Annapurna Labs

Do you love decomposing problems to develop products that impact millions of people around the world? Would you enjoy identifying, defining, and building software solutions that revolutionize how businesses operate? The Annapurna Labs team at Amazon Web Services (AWS) is looking for a Software Development Engineer II to build, deliver, and maintain complex products that delight our customers and raise our performance bar.

As a deep learning compiler engineer on the Neuron team, you will be a thought leader supporting the development of a compiler targeting AWS Inferentia and Trainum. You will be developing and scaling the compiler to handle the world's largest ML workloads. You'll design fault-tolerant systems that run at massive scale as we continue to innovate best-in-class services and applications in the AWS Cloud.

Key responsibilities:

  • Solve challenging technical problems at every layer of the stack
  • Design, implement, test, deploy, and maintain innovative software solutions
  • Build high-quality, highly available, always-on products
  • Research implementations that deliver the best possible experiences for customers
  • Create metrics, implement automation and other improvements
  • Participate in design discussions, code review, and communicate with stakeholders
  • Work cross-functionally to help drive business decisions with technical input

AWS values diverse experiences and encourages candidates from all backgrounds to apply. The team puts a high value on work-life balance and offers flexibility in working hours. There are opportunities for mentorship and career growth within a supportive team environment.

Required Qualifications:

  • 3+ years of non-internship professional software development experience
  • 2+ years of non-internship design or architecture experience
  • Experience programming with at least one software programming language

Preferred Qualifications:

  • 3+ years of full software development life cycle experience
  • Bachelor's degree in computer science or equivalent

Amazon is an equal opportunity employer committed to a diverse and inclusive workplace.

Last updated a month ago

Responsibilities For ML Compiler Engineer, AWS Neuron, Annapurna Labs

  • Develop and scale compilers for large ML workloads
  • Design fault-tolerant systems at massive scale
  • Solve challenging technical problems across the stack
  • Design, implement, test, deploy, and maintain innovative software solutions
  • Build high-quality, highly available products
  • Research implementations for best customer experiences
  • Create metrics and implement automation
  • Participate in design discussions and code reviews
  • Communicate with internal and external stakeholders
  • Work cross-functionally to drive business decisions with technical input

Requirements For ML Compiler Engineer, AWS Neuron, Annapurna Labs

Python
  • 3+ years of non-internship professional software development experience
  • 2+ years of non-internship design or architecture experience
  • Experience programming with at least one software programming language
  • Bachelor's degree in computer science or equivalent (preferred)
  • 3+ years of full software development life cycle experience (preferred)

Benefits For ML Compiler Engineer, AWS Neuron, Annapurna Labs

  • Flexible working hours
  • Mentorship opportunities
  • Career growth opportunities

Interested in this job?