Sr. Deep Learning Compiler Engineer III, AWS Neuron, Annapurna Labs

Amazon Web Services (AWS) is the world's most comprehensive and broadly adopted cloud platform, pioneering cloud computing and continuous innovation.
Machine Learning
Staff Software Engineer
In-Person
5,000+ Employees
5+ years of experience
AI · Enterprise SaaS

Description For Sr. Deep Learning Compiler Engineer III, AWS Neuron, Annapurna Labs

Join the innovative Annapurna Labs team at AWS as a Senior Deep Learning Compiler Engineer, where you'll work on cutting-edge products at the intersection of machine learning, high-performance computing, and distributed architectures. AWS Neuron is a groundbreaking SDK that optimizes neural net model performance on AWS Inferentia and Trainium custom chips. You'll be part of a team democratizing access to cutting-edge ML infrastructure, working with technologies like TensorFlow, PyTorch, and MXNet.

As a senior engineer, you'll architect and implement critical features, publish research, and mentor experienced engineers. The role combines technical leadership with hands-on development, requiring expertise in compiler development and deep learning systems. You'll work in an agile, startup-like environment where innovation is constant and impact is immediate.

AWS offers an inclusive culture with 10 employee-led affinity groups and various benefits supporting work-life balance. The team values knowledge sharing, mentorship, and career growth, assigning projects that help members develop into well-rounded professionals. AWS embraces diverse experiences and backgrounds, encouraging applications from candidates with non-traditional paths.

Key responsibilities include solving complex technical challenges, designing scalable solutions, and collaborating across teams to drive business decisions. You'll work on high-impact projects that directly affect AWS's global customer base, while maintaining high standards for software quality and reliability.

The position offers the opportunity to work with cutting-edge ML infrastructure, contribute to innovative products, and shape the future of cloud computing. You'll be part of AWS's culture of inclusion, guided by their 16 Leadership Principles, which emphasize seeking diverse perspectives, continuous learning, and earning trust.

Last updated 12 days ago

Responsibilities For Sr. Deep Learning Compiler Engineer III, AWS Neuron, Annapurna Labs

  • Architect and implement business-critical features
  • Develop and scale compiler targeting AWS Inferentia and Trainum
  • Support development of compiler for handling large ML workloads
  • Participate in pre-silicon design
  • Bring new products/features to market
  • Solve challenging technical problems across all stack layers
  • Design, implement, test, deploy and maintain innovative software solutions
  • Build high-quality, highly available products
  • Mentor and lead engineering team

Requirements For Sr. Deep Learning Compiler Engineer III, AWS Neuron, Annapurna Labs

Python
  • 5+ years of non-internship professional software development experience
  • 5+ years of programming with at least one software programming language
  • 5+ years of leading design or architecture experience
  • Experience as a mentor, tech lead or leading an engineering team
  • Experience with full software development life cycle
  • Bachelor's degree in computer science or equivalent

Benefits For Sr. Deep Learning Compiler Engineer III, AWS Neuron, Annapurna Labs

  • Work-life balance
  • Flexible working hours
  • Mentorship & Career Growth
  • Learning and development opportunities

Interested in this job?

Jobs Related To Amazon Sr. Deep Learning Compiler Engineer III, AWS Neuron, Annapurna Labs

Sr SOC Verification Engineer, Annapurna ML

Senior SOC Verification Engineer role at AWS, focusing on custom chip design verification for cloud-scale machine learning acceleration.

Sr. Amazon Q Delivery Engineer, Amazon Q Customer Success Team (Q-CST)

Senior Amazon Q Delivery Engineer position focusing on implementing enterprise-scale Generative AI solutions using Amazon Q and AWS AI/ML services, requiring 7+ years of cloud architecture experience.

Senior SoC Architectural Modeling Engineer, AWS Machine Learning Accelerators

Senior SoC Modeling Engineer position at AWS, developing functional models for machine learning accelerator chips, requiring expertise in C++/SystemC and hardware modeling.

Sr ML Compiler Engineer, Annapurna Labs

Senior Machine Learning Compiler Engineer position at Amazon's Annapurna Labs focusing on ML accelerator development and optimization

Sr. Machine Learning - Compiler Engineer III, AWS Neuron, Annapurna Labs

Senior Machine Learning Compiler Engineer position at AWS Neuron team, focusing on optimizing ML models for AWS Inferentia and Trainium custom chips.