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

World's most comprehensive and broadly adopted cloud platform, pioneering cloud computing and continuous innovation.
$150,000 - $250,000
Machine Learning
Senior 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

AWS Neuron team is seeking a Senior Deep Learning Compiler Engineer to join Annapurna Labs, focusing on revolutionizing deep learning accessibility. The role involves working with AWS Neuron SDK, which optimizes neural net models' performance on AWS Inferentia and Trainium custom chips. You'll be part of a team developing cutting-edge compiler technology for machine learning workloads, working at the intersection of ML, high-performance computing, and distributed architectures. The position offers unique opportunities to influence AWS's machine learning infrastructure, collaborate with diverse teams, and solve complex technical challenges. The team operates in a startup-like environment while backed by AWS's robust infrastructure and resources. You'll contribute to pre-silicon design, product development, and innovation in ML acceleration. AWS provides an inclusive culture, emphasizing work-life balance, mentorship, and professional growth. The role combines technical leadership with hands-on development, requiring expertise in compiler development and an understanding of machine learning systems. You'll work with AWS ML services teams, contribute to product strategy, and help shape the future of machine learning infrastructure at AWS.

Last updated 2 months ago

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

  • Design fault-tolerant systems that run at massive scale
  • Architect and implement business-critical features
  • Publish cutting-edge research
  • Mentor a team of experienced engineers
  • Develop and scale compiler targeting AWS Inferentia and Trainum
  • Support pre-silicon design
  • Bring new products/features to market

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 preferred
  • Bachelor's degree in computer science or equivalent preferred

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

  • Work-life balance
  • Flexible working hours
  • Mentorship and career growth opportunities
  • Inclusive team culture
  • Employee-led affinity groups
  • Learning experiences and conferences

Interested in this job?

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

Sr. Software Engineer- AI/ML, AWS Neuron Distributed Training

Senior ML Engineer role at AWS focusing on distributed training systems and ML accelerators, offering competitive compensation and growth opportunities.

Senior Software Engineer, AWS Neuron Inference

Senior Software Engineer position at Amazon working on AWS Neuron ML inference optimization and development.

Senior Machine Learning Engineer, Generative AI, AGI Inference Engine

Senior ML Engineer role focusing on optimizing inference for LLMs and diffusion models, leading research and implementation of novel techniques in Generative AI at Amazon's AGI team.

Software Development Engineer, Prime Video Search

Senior Software Engineer role at Amazon Prime Video focusing on machine learning and search infrastructure development.

Sr. Machine Learning Engineer, Amazon Q in QuickSight

Senior Machine Learning Engineer position at Amazon working on Q in QuickSight, focusing on LLMs and NLP for business intelligence solutions.