Senior Machine Learning Compiler Engineer, AWS Neuron, Annapurna Labs

Job overview

The Senior Machine Learning Compiler Engineer, AWS Neuron, Annapurna Labs role is an onsite position based in Cupertino, CA. Compensation is not specified, though the role supports the AWS Neuron compiler team. This role exists to build the next-generation Neuron compiler, transforming ML models from frameworks like PyTorch and TensorFlow for deployment on AWS Inferentia and Trainium chips. The engineer will solve complex optimization problems to achieve peak performance for large-scale AI models.

What we're looking for

Skills & competencies

full-time

day shift

cupertino ca

annapurna labs

senior engineer

machine learning

compiler

aws neuron

c++

python

onsite

no on-call

no travel

Work arrangement

• Weekend coverage required

Benefits & perks

• Health insurance, 401(k) matching, paid time off, parental leave, sign-on payments, restricted stock units (RSUs)

About the employer

Annapurna Labs (U.S.) Inc. is hiring for this role. Industry: Custom Computer Programming Services. Sector: 54.