Machine Learning Compiler Engineer

ML Compiler Engineer 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. 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 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 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. 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.