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

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 day 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?

Jobs Related To Amazon ML Compiler Engineer, AWS Neuron, Annapurna Labs

Conversational Engineer

66degrees seeks a Conversational Engineer to build complex virtual agents using Dialogflow CX, lead workshops, and collaborate with cross-functional teams.

AI Application Engineer

AI Application Engineer at DoorDash: Drive innovation in People Applications using cutting-edge AI/ML solutions.

Machine Learning Engineer 2

Join Adobe's Document Cloud team as a Machine Learning Engineer to develop innovative AI/ML solutions for processing PDFs at scale.

Software Developer 3

Oracle is seeking a Software Developer 3 to join their OCI Data Science team, focusing on machine learning and cloud infrastructure.

ML Engineer

ML Engineer role at Beam, developing AI-Agents with focus on prompt engineering and model fine-tuning.