Drimva

Qualcomm

Senior Machine Learning Engineer – AI/ML Compiler

$151K–$227K

Santa Clara, California · On-site · Senior · Full-time · Posted Sep 17, 2026

Experience
Bachelor's: 2+ years of Hardware Engineering, Software Engineering, Systems Engineering, or related work experience; master's: 1+ year; PhD: no experience stated. 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 related field.

About the job

Build and maintain AI/ML compiler tools that optimize and validate models for efficient execution on device CPUs, GPUs, and NPUs.

What you'll do

  • Develop compiler technologies that convert PyTorch or ONNX models into efficient device code.
  • Contribute to AI hub compiler and ONNX Runtime QNN, including graph optimization and backend partitioning.
  • Build debugging tools to diagnose compilation failures, accuracy loss, and performance issues.

What they ask for

  • Bachelor's degree in a related field plus 2+ years of relevant experience; or master's degree plus 1+ year; or PhD.
  • Proficient 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.

Nice to have

  • Experience with PyTorch model export and on-device deployment frameworks.
  • Familiarity with SoC constraints and hardware-specific runtimes such as QAIRT/QNN; CI/CD pipeline experience.

Pay

$151,200.00 - $226,800.00 USD base salary

Tools

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

AI focus

Develop and optimize ML compilers and tools for deploying AI models on device chips.

Model-written summary · the original posting is authoritative

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