Drimva

Qualcomm

Staff Machine Learning Engineer – AI/ML Compiler

$174K–$261K

Santa Clara, California +1 · On-site · Staff · Full-time · Posted Sep 17, 2026

Location
Santa Clara, California
San Diego, California
Experience
4+ years with a bachelor’s degree, 3+ years with a master’s degree, or 2+ years with a PhD; preferred: 3+ years in ML infrastructure, compiler engineering, or AI framework development.
Education
Bachelor’s, master’s, or PhD in Computer Science, Engineering, Information Systems, or a related field.

About the job

Own and develop Qualcomm AI Hub’s model compilation infrastructure, enabling models to be converted, validated, and optimized for Qualcomm devices and runtimes.

What you'll do

  • Develop compilation paths and pipeline infrastructure for PyTorch and ONNX models, including graph transformations, validation, and backend dispatch.
  • Build tools and automated workflows to compile, profile, debug, and validate models across supported runtimes and Qualcomm devices.
  • Collaborate with internal teams and model partners, and create technical documentation and examples for developers.

What they ask for

  • Bachelor’s degree in a related field and 4+ years of relevant work experience; or a master’s degree and 3+ years; or a PhD and 2+ years.
  • Proficiency in Python and C++.
  • Understanding of ML compiler concepts and hands-on experience with one or more compiler stacks such as MLIR, ONNX, or TVM.
  • Experience with PyTorch model export and on-device deployment frameworks.

Nice to have

  • Familiarity with SoC constraints and hardware-specific runtimes such as QAIRT/QNN.
  • Experience building automated CI/CD pipelines for model compilation and validation at scale.

Pay

$174,000.00 - $261,000.00 USD base salary

Tools

Python, C++, PyTorch, ONNX, MLIR, TVM, LiteRT, ONNXRuntime

AI focus

Develop infrastructure to compile, optimize, validate, and profile ML models for on-device deployment across CPU, GPU, and NPU backends.

Model-written summary · the original posting is authoritative

First seen Oct 28, 2026 · Last verified Nov 5, 2026 (UTC)