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

Job Description

Senior Software Engineer role in the Compiler team for AWS Neuron, focusing on optimizing ML models for AWS Inferentia and Trainium servers.

Requirements

• 5+ years of non-internship professional software development experience
• Experience in object-oriented languages like C++/Java
• Experience with compilers or building ML models using ML frameworks on accelerators (e.g., GPUs) is preferred
• Experience with OpenSource compiler toolset like LLVM/MLIR
• Master's degree or PhD in Computer Science, or a related technical field

Responsibilities

• Design, implement, test, deploy and maintain innovative software solutions for Neuron compiler's performance and stability
• Work with chip architects, runtime/OS engineers, scientists and ML Apps teams to deploy ML models on AWS accelerators
• Become a front-face of Neuron Compiler to work with open-source communities and influence industry partners
• Build features that deliver best possible experiences for customers

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)