Senior Machine Learning Engineer – AI/ML Compiler

Teknik, data och digitalt · Data, AI och analys · Maskininlärning · Mjukvaruutveckling

I korthet

Qualcomm is seeking a Senior Machine Learning Engineer specializing in AI/ML Compiler technology. This role involves building and maintaining machine learning compiler technologies that optimize AI models for on-device chips. You will contribute to AI hub compilers and optimize ML frameworks, ensuring efficient model execution and clear diagnostics.

Ansvarsområden

  • Build and maintain machine learning compiler technologies that turn AI models (from PyTorch or ONNX) into efficient code that runs on device chips (CPU, GPU, and NPU processors).
  • Contribute to AI hub compiler, ONNX Runtime QNN —doing graph optimization, partitioning, and making sure models work correctly across backends.
  • Build debugging tools to spot and fix failures, accuracy loss, or slowdowns, with clear diagnostics for other developers.
  • Solve open-ended problems independently while mentoring teammates and giving technical guidance.
  • Explain complex compiler ideas clearly to chip engineers, business partners, and outside developers.

Krav

  • Bachelor's degree in Computer Science, Engineering, Information Systems, or related field and 2+ years of Hardware Engineering, Software Engineering, Systems Engineering, or related work experience.
  • OR Master's degree in Computer Science, Engineering, Information Systems, or related field and 1+ year of Hardware Engineering, Software Engineering, Systems Engineering, or related work experience.
  • OR PhD in Computer Science, Engineering, Information Systems, or related field.
  • Proficient in Python and C++
  • Solid understanding of ML compiler concepts (graph IRs, operator fusion, shape inference, lowering passes, backend partitioning) and hands-on experience with one or more compiler stacks such as MLIR, ONNX, or TVM
  • Experience with PyTorch model export (torch.export, torch.compile, FX, ATen IR)
  • Experience with on-device deployment frameworks such as LiteRT, ExecuTorch, or ONNXRuntime
  • Strong written and verbal communication skills
  • Proficiency with git and software engineering best practices

Önskade kvalifikationer

  • 3+ years of industry experience in ML infrastructure, compiler engineering, or AI framework development
  • Familiarity with SoC-level constraints (memory bandwidth, compute precision, NPU/DSP execution) and hardware-specific runtimes such as QAIRT/QNN is a plus
  • Experience building automated CI/CD pipelines for model compilation and validation at scale

Förmåner

  • Competitive annual discretionary bonus program
  • Opportunity for annual RSU grants
  • Highly competitive benefits package designed to support your success at work, at home, and at play.
#Machine Learning#AI#Compiler Engineering#ML Infrastructure#PyTorch#ONNX#MLIR#TVM#C++#Python
Qualcomm Logo

Företag

Qualcomm

Publicerade jobb

för 1 vecka sedan

Anställningstyp

Heltid

Arbetsform

På plats

Erfarenhetsnivå

Senior

Platser

Santa Clara, United States of America

Kvalifikation

Kandidatexamen, Masterexamen, Doktorand

Sökande

Ansök tidigt