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