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

Job Description

AWS Machine Learning Compiler Engineer III at Amazon Annapurna Labs team responsible for silicon development at AWS. Working on AWS Neuron team to optimize performance of neural net models on custom-built AWS hardware.

Requirements

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

Responsibilities

• Support the ground-up development and scaling of a compiler for ML workloads
• Architect and implement business-critical features
• Publish cutting-edge research
• Mentor a team of experienced engineers
• Work on pre-silicon design and new product/feature development

Benefits

• Medical, dental, and vision coverage (multiple plan options)
• Health savings and flexible spending accounts
• 24/7 Employee Assistance Program for mental health support
• 401(k) with company match and various investment choices
• Company-paid life/AD&D and disability insurance
• Restricted stock units (RSUs) for employee ownership
• Child care and elder care referrals, adoption assistance
• Paid maternity and parental leave (including Ramp Back)
• Paid time off (PTO) and company holidays
• Employee discount on Amazon.com products
• Career Choice program (95% tuition and textbook coverage)