GPU Compute & MLIR Compiler Engineer — AI Workloads

About the role

GPU Compute & MLIR Compiler Engineer — AI Workloads at Qualcomm. Develop and optimize GPU compute kernels targeting OpenCL and Vulkan compute backends for high-throughput AI/ML workloads.

What you'll do

• Develop and optimize GPU compute kernels targeting OpenCL and Vulkan compute backends for high-throughput AI/ML workloads
• Design, build, and extend MLIR dialects across multiple abstraction levels — including frontend dialects, graph-level IR, tensor IR (e.g., Linalg, Tensor, Tosa)
• Implement and maintain MLIR-based compiler passes and transformations, including tiling, fusion, bufferization, vectorization
• Conduct profiling and bottleneck analysis of compiled kernels using GPU counters and vendor-specific profilers
• Build and maintain GPU runtime infrastructure for both OpenCL and Vulkan, including memory management, pipeline setup, command buffer orchestration

What you need

• 4–7 yrs of experience
• Machine learning
• LLMs
• C++
• C
• Computer vision
• B.Tech or B.E. or B.Sc or BCA or M.Tech or PhD