Taro Logo

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 2 months 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