Machine Learning - Compiler Engineer, Annapurna Labs

Job Description

As a Machine Learning Compiler Engineer II on the AWS Neuron team, you will be responsible for developing and scaling a compiler to handle large ML workloads on AWS hardware.

Requirements

• 3+ years of non-internship professional software development experience
• 2+ years of experience architecting and optimizing compilers
• Proficiency with 1 or more of the following programming languages: C++ (preferred), C, Python

Responsibilities

• Support ground-up development and scaling of a compiler for large ML workloads
• Architect and implement business-critical features
• Contribute to a team of experienced engineers
• Leverage technical communication skills with AWS ML services teams

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)