Staff Machine Learning Engineer – AI/ML Compiler

Qualcomm

Actively hiring
Santa Clara, CA · San Diego, CA Posted 82 days ago $160,500–$240,700 / year
View original post Log in to save

At a glance

AI generated

TL;DR

Join the Qualcomm AI Hub Compiler team as a senior engineer responsible for designing and maintaining the end-to-end compilation pipeline that powers model conversions from PyTorch or ONNX to deployable artifacts on Snapdragon SoCs using LiteRT, ONNXRuntime, or QAIRT. You will develop ONNX-based graph transformation passes, handle opset compatibility, and contribute to ONNXRuntime QNN execution provider optimizations. Additionally, you will build tooling for analyzing compilation issues and author documentation for the developer community. Proficiency in Python and C++, along with experience in ML compiler concepts like graph IRs and shape inference, is essential. Familiarity with SoC constraints and hardware-specific runtimes such as QAIRT/QNN is beneficial. This role involves significant influence on the Qualcomm AI Hub's compiler roadmap and model catalog strategy, impacting developer experiences across various use cases including vision, audio, speech, and multi-modal models.

Skills

Python C++ ONNX PyTorch MLIR LiteRT ONNXRuntime CI/CD Git ATen QNN QAIRT TVM Prometheus Grafana

What you'll do

  • Design and maintain end-to-end compilation pipeline for Qualcomm AI Hub Workbench.
  • Build ONNX-based compilation paths including graph transformation passes and op validation.
  • Develop PyTorch compilation paths to handle dynamic shapes and custom operations.
  • Contribute to ONNXRuntime QNN execution provider with optimizations and validations.
  • Create tooling for analyzing, profiling, and debugging model compilation issues.

What we're looking for

  • 3+ years of industry experience in ML infrastructure, compiler engineering, or AI framework development.
  • Proficient in Python and C++ programming languages.
  • Solid understanding of ML compiler concepts including graph IRs, operator fusion, shape inference, lowering passes, and backend partitioning.
  • Experience with PyTorch model export and on-device deployment frameworks like LiteRT, ONNXRuntime.
  • Familiarity with SoC-level constraints and hardware-specific runtimes such as QAIRT/QNN.
  • Strong written and verbal communication skills; proficiency with git and software engineering best practices.
  • Experience building automated CI/CD pipelines for model compilation and validation at scale.

Market check

Salary context

This $160,500–$240,700 range sits above 42% of similar postings on FindRole.

Peer median band

$161,700–$260,550

Median floor and ceiling across peers.

Typical midpoint (25–75%)

$183,650–$249,753

Middle half of comparable postings.

Based on 240 comparable postings.

* 240 is the maximum number of comparable postings sampled.

Employer

About Qualcomm

Qualcomm is a leading American semiconductor and telecommunications company based in San Diego, CA.

Qualcomm currently has 595 open roles on FindRole.

Listed pay typically runs $148,300–$222,500 across 540 roles with salary data.

Most-posted roles

View all roles at Qualcomm

More like this

Similar roles

Staff Machine Learning Engineer

Intuit

Mountain View, CA 49 days ago $202,500–$274,000
Python Scikit-learn NLTK NumPy Pandas TensorFlow Keras R Spark SQL Git AWS GCP CI/CD

Staff Machine Learning Engineer

Arm Holdings

Austin, TX 49 days ago $249,900–$338,100
Python TensorFlow PyTorch GPU ARM ML Model Optimization Deep Learning Computer Architecture CI/CD
Hybrid

Staff Machine Learning Engineer

Intuit

Mountain View, California 45 days ago $197,000–$266,500
Python Scikit-learn NLTK NumPy Pandas TensorFlow Keras R Spark SQL Git AWS GCP CUDA cuDNN