Machine Learning Compiler Engineer

Company:

Job Area:

General Summary:

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

The Compiler Labs unit in Qualcomm’s 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.

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

What makes this work exciting right now

The complexity of modern ML architectures demands multi-tier compilers and new thinking about phase ordering. Polyhedral optimization is directly relevant to the deeply nested imperfect loop nests that characterize generative AI - and the implied execution model is pulling polyhedral techniques toward dataflow concepts and structured speculation. Communication-avoiding algorithms from high-performance computing are increasingly relevant. We are advancing compilers and intermediate representations that generalize cleanly across new innovations in ML model algorithms, so that our work compounds rather than restarts. We are working on correctness verification for programs built around quantized neural operators. And we are contributing to open-source dialect convergence in the LLVM and MLIR ecosystems - we have open-sourced our Hexagon-MLIR compiler and are actively developing it for Triton and Torch-MLIR flows.

Generative AI is part of the work

Generative AI is reshaping compiler development itself - both as a tool that accelerates how engineers build compilers, and as a capability that lives inside compilers. Domain-specific languages are a particularly strong target for AI tools in compilation, because the compression, abstraction, and structure DSLs provide give AI models the right level of representation to operate on. We expect our engineers to be fluent in incorporating generative AI into the compiler development workflow and into compilers themselves.

Our technical approach

Our approach 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 keen about seeing our compilers used and having large impact on Qualcomm’s business.

Levels and locations

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 to entry-level engineers.

We are looking for candidates to join our teams in New York City, Austin, San Diego, and Raleigh, NC. For experienced candidates, we may consider remote work arrangements.

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
• Incorporate generative AI into the compiler development workflow and into compilers themselves
• 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

• 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
• Experience incorporating generative AI into compiler development workflow (AI-assisted code generation, AI-assisted analysis, AI-assisted refactoring and review)
• Experience designing or extending DSLs as targets for AI-driven program synthesis
• ML applications and ML optimization experience
• ML architecture experience
• High-performance computing experience, including communication-avoiding algorithms
• Polyhedral compiler optimization experience
• Loop transformation and vectorization experience
• GPU programming, parallel programming experience
• General optimization experience

Minimum Qualifications:

OR

Master's degree in Computer Science, Engineering, Information Systems, or related field and 7+ years of Hardware Engineering, Software Engineering, Systems Engineering, or related work experience.

OR

PhD in Computer Science, Engineering, Information Systems, or related field and 6+ years of Hardware Engineering, Software Engineering, Systems Engineering, or related work experience.

Qualcomm is an equal opportunity employer. If you are an individual with a disability and need an accommodation during the application/hiring process, rest assured that Qualcomm is committed to providing an accessible process. You may e-mail [email protected] or call Qualcomm's toll-free number found here. Upon request, Qualcomm will provide reasonable accommodations to support individuals with disabilities to be able participate in the hiring process. Qualcomm is also committed to making our workplace accessible for individuals with disabilities. (Keep in mind that this email address is used to provide reasonable accommodations for individuals with disabilities. We will not respond here to requests for updates on applications or resume inquiries).

To all Staffing and Recruiting Agencies:Our Careers Site is only for individuals seeking a job at Qualcomm. Staffing and recruiting agencies and individuals being represented by an agency are not authorized to use this site or to submit profiles, applications or resumes, and any such submissions will be considered unsolicited. Qualcomm does not accept unsolicited resumes or applications from agencies. Please do not forward resumes to our jobs alias, Qualcomm employees or any other company location. Qualcomm is not responsible for any fees related to unsolicited resumes/applications.

EEO Employer: Qualcomm is an equal opportunity employer; all qualified applicants will receive consideration for employment without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, disability, Veteran status, or any other protected classification.

Qualcomm expects its employees to abide by all applicable policies and procedures, including but not limited to security and other requirements regarding protection of Company confidential information and other confidential and/or proprietary information, to the extent those requirements are permissible under applicable law.

Pay rangeand Other Compensation & Benefits :

The above pay scale reflects the broad, minimum to maximum, pay scale for this job code for the location for which it has been posted. Even more importantly, please note that salary is only one component of total compensation at Qualcomm. We also offer a competitive annual discretionary bonus program and opportunity for annual RSU grants (employees on sales-incentive plans are not eligible for our annual bonus). In addition, our highly competitive benefits package is designed to support your success at work, at home, and at play. Your recruiter will be happy to discuss all that Qualcomm has to offer - and you can review more details about our US benefits at this link.

If you would like more information about this role, please contact Qualcomm Careers.