Machine Learning - Compiler Engineer II, Annapurna Labs

Amazon • Boston, MA

1d • Onsite

About The Position

The AWS Neuron team is developing a deep learning compiler stack that takes neural network descriptions created in frameworks such as TensorFlow, PyTorch, and MXNET, and converts them into code suitable for execution. This role involves supporting the ground-up development and scaling of a compiler to handle the world's largest ML workloads. The engineer will be responsible for architecting and implementing business-critical features, publishing cutting-edge research, and contributing to a team of experienced engineers. This role will also leverage technical communication skills as a hands-on partner to AWS ML services teams and will be involved in pre-silicon design, bringing new products/features to market, and other projects.

Requirements

• 3+ years of non-internship professional software development experience
• 2+ years of non-internship design or architecture (design patterns, reliability and scaling) of new and existing systems experience
• Experience programming with at least one software programming language
• 3+ years of non-internship professional software development, or 3+ years of 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

Nice To Haves

• A background in Machine Learning and AI accelerators is preferred, but not required.
• 3+ years of full software development life cycle, including coding standards, code reviews, source control management, build processes, testing, and operations experience
• Bachelor's degree in computer science or equivalent
• PhD in computer science, computer engineering, or related field, or MS degree
• Experience with multiple toolchains and Instruction Set Architectures
• Proficiency with resource management, scheduling, code generation, and compute graph optimization
• Experience optimizing Tensorflow, PyTorch or MxNET deep learning models

Responsibilities

• Supporting the ground-up development and scaling of a compiler to handle the world's largest ML workloads.
• Architecting and implementing business-critical features.
• Publishing cutting-edge research.
• Contributing to a brilliant team of experienced engineers.
• Leveraging technical communication skills as a hands-on partner to AWS ML services teams.
• Being involved in pre-silicon design.
• Bringing new products/features to market.

Benefits

• health insurance (medical, dental, vision, prescription, Basic Life & AD&D insurance and option for Supplemental life plans, EAP, Mental Health Support, Medical Advice Line, Flexible Spending Accounts, Adoption and Surrogacy Reimbursement coverage)
• 401(k) matching
• paid time off
• parental leave
• sign-on payments
• restricted stock units (RSUs)

What This Job Offers

• Job Type: Full-time
• Career Level: Mid Level
• Education Level: Associate degree