ML/AI Compiler Engineer: Frameworks & Performance

QUALCOMM, Inc.; Markham; On-site; CAD 120,000 - 190,000; Full time

Job summary

Qualcomm Canada ULC is advancing on-device AI by developing ML/AI compilers for efficient mapping of workloads on Qualcomm hardware. You will contribute to compiler development, optimize performance and power, and study ML workloads to inform future HW/SW features, with a focus on PyTorch integration and cross-functional collaboration.

The role requires a strong background in algorithms, performance analysis, and programming in C++/Python, with experience in ML frameworks and compiler flows.

Qualifications

• Bachelor's degree or higher in CS, Engineering, or related field.
• Strong background in algorithms and performance analysis.
• Experience with ML frameworks and compiler/toolchains.

Responsibilities

• Develop and evolve ML/AI compilers for efficient mappings of ML/AI workloads on Qualcomm hardware.
• Analyze ML workloads to guide future HW/SW features.
• Create algorithms to couple PyTorch with Qualcomm compiler flows.
• Study ML model trends to inform design of software and hardware.

Skills

C++

Python

ML algorithms

Performance analysis

Software engineering

Education

Bachelor's degree in CS or related

Master's degree in CS or related

PhD in CS or related

Tools

Git

Docker

ONNX

PyTorch

Job description

Qualcomm Canada ULC is advancing on-device AI by developing ML/AI compilers for efficient mapping of workloads on Qualcomm hardware. You will contribute to compiler development, optimize performance and power, and study ML workloads to inform future HW/SW features, with a focus on PyTorch integration and cross-functional collaboration.

The role requires a strong background in algorithms, performance analysis, and programming in C++/Python, with experience in ML frameworks and compiler flows.