GPU Compute & MLIR Compiler Engineer — AI Workloads
BengaluruFull-time4–7 yrsPosted over a month ago
Mid-level AI role at Qualcomm in Bengaluru. Needs 4+ years and a B.Tech, B.E. or B.Sc.
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
- Machine learning
- LLMs
- C++
- C
- Computer vision