Machine Learning - Compiler Engineer II, Annapurna Labs

About the position

As a Machine Learning Compiler Engineer II on the AWS Neuron team, you will be 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, and contributing to a brilliant team of experienced engineers excites and challenges you. You will leverage your technical communications skill as a hands-on partner to AWS ML services teams and you will be involved in pre-silicon design, bringing new products/features to market, and many other exciting projects. A background in Machine Learning and AI accelerators is preferred, but not required.

Responsibilities

• Support the development and scaling of a compiler for large ML workloads.
• Architect and implement business-critical features.
• Publish cutting-edge research.
• Collaborate with AWS ML services teams.
• Participate in pre-silicon design and product feature launches.

Requirements

• 3+ years of non-internship professional software development experience.
• 2+ years of experience architecting and optimizing compilers.
• Proficiency with C++ (preferred), C, or Python.

Nice-to-haves

• M.S. or Ph.D. in Computer Science or related field.
• 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.

Benefits

• Flexible working hours.
• Mentorship and career growth opportunities.
• Inclusive team culture with employee-led affinity groups.
• Commitment to work-life balance.