Staff ML Engineer — On-Device AI Compiler Lead

Qualcomm; San Diego (CA); On-site; USD 160,500 - 240,700; Full time

Benefits offered by this job

Annual discretionary bonus program

Opportunity for RSU grants

Comprehensive benefits package

Job summary

A leading technology company is seeking an experienced Machine Learning Engineer to own the infrastructure for model compilations within the Qualcomm AI Hub. You will design and maintain the end-to-end compilation pipeline, ensuring efficient model execution across various backends. Candidates should have a relevant degree and substantial experience in compiler engineering, ML infrastructure, and proficiency in Python and C++. A competitive salary range of $160,500 to $240,700 is offered, alongside a comprehensive benefits package.

Qualifications

• Bachelor's, Master's, or PhD in relevant field with appropriate work experience.
• 3+ years of industry experience in ML infrastructure or compiler engineering.
• Proficiency in Python and C++.

Responsibilities

• Design, develop, and maintain the compilation pipeline for Qualcomm AI Hub Workbench.
• Build automated compilation pipelines and CI/CD evaluation harnesses.
• Collaborate with internal teams for model onboarding.

Skills

Compiler engineering

Python

C++

Machine Learning infrastructure

Graph IRs

Automation

Technical documentation

Education

Bachelor's degree in Computer Science, Engineering, Information Systems, or related field

Master's degree in Computer Science, Engineering, Information Systems or related field

PhD in Computer Science, Engineering, Information Systems, or related field

Tools

PyTorch

ONNX

CI/CD pipelines

Job description

A leading technology company is seeking an experienced Machine Learning Engineer to own the infrastructure for model compilations within the Qualcomm AI Hub. You will design and maintain the end-to-end compilation pipeline, ensuring efficient model execution across various backends. Candidates should have a relevant degree and substantial experience in compiler engineering, ML infrastructure, and proficiency in Python and C++. A competitive salary range of $160,500 to $240,700 is offered, alongside a comprehensive benefits package.