Engineer, Staff

If you’re interested in advancing and applying mathematics, programming languages theory, and advanced algorithms to program optimization for cutting-edge machine learning accelerators, then you really want to be talking to us!

We are looking to hire ML Compiler engineers to join our team. We work tactically on improving existing ML compilers and strategically on developing new and innovative ML compilers.

Our technical approach to compilers emphasizes powerful representations for precisely and compactly modeling programs and the optimization challenges and using advanced mathematics and algorithms for performing optimizations.

We are also solid in using “old school” compiler technologies as they apply to contemporary ML challenges, and in meticulous software engineering to produce beautiful compilers. We are also keen about seeing our compilers used and having large impacts on Qualcomm’s business.

Mapping ML algorithms to ML accelerators is currently one of the most interesting and challenging problems for compilers. Our compiler targets include the Qualcomm Neural Signal Processor, Adreno GPUs, low-power ML accelerators, and CPU accelerators.

This job description spans multiple levels, from entry to experienced. Our team is a good home for compiler developers with advanced degrees, and we have solid mentoring and give substantial responsibility quickly for entry level engineers.

Responsibilities
Work on a wide range of ML compilers

Improve ML compiler optimization capabilities through benchmark analysis and profiling

Innovate new ML compiler and optimization algorithms

Upstream compiler algorithms to open-source compiler projects
Author research publications and represent the company in conferences and industry forums

Required

Experience with compiler development and computer architecture

ML experience

A degree in the field of computer science or applied mathematics

Experience with software engineering

Solid intellectual ability, motivation, and a strong history of achievement
Excellent oral and written communication skills

Desired
Experience with MLIR, MLIR Dialects (LinAlg, Affine), Pytorch 2.0, TVM, Triton, and/or LLVM

SYCL experience

ML applications and ML optimization experience

ML architecture experience

High performance computing experience

Polyhedral compiler optimization experience

Loop transformation and vectorization experience

GPU programming, parallel programming experience

General optimization experience

8+ years of relevant work experience