Machine Learning Compiler Engineer

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!

The Compiler Labs unit in Qualcomm AI Software department is looking for 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.

Key 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 • For senior levels - Lead and manage projects while doing substantial technical work

Required Qualifications: • Bachelor's degree in Computer Science, Engineering, Information Systems, or related field and 2+ years of relevant work experience, OR • Master's degree in Computer Science, Engineering, Information Systems, or related field and 1+ year of relevant work experience, OR • PhD in Computer Science, Engineering, Information Systems, or related field

Preferred Qualifications: • Master's degree in Computer Science, Engineering, Electrical Engineering, or related field • Experience with compiler development and computer architecture • ML experience • Experience with software engineering • Solid intellectual ability, motivation, and a strong history of achievement • Excellent oral and written communication skills • 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

Qualcomm offers competitive salaries, excellent benefits, and a supportive work environment focused on innovation and career growth.