ML Compiler Engineer, Annapurna Labs

Job Description

Join the AWS Neuron Compiler team as a skilled compiler engineer to develop a deep learning compiler stack for optimizing application models across diverse domains.

Requirements

• B.S. or M.S. in computer science or related field
• Proficiency with C++ or Python
• 3+ years of non-internship professional software development experience
• 2+ years of experience in compiler optimization, graph-theory, hardware bring-up, etc.

Responsibilities

• Design, implement, test, deploy, and maintain innovative software solutions
• Research implementations for enhancing customer experiences
• Collaborate across diverse teams to solve technical challenges

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)