GPU Compute & MLIR Engineer

This role is for one of Weekday’s clients
Salary range: Rs 2000000 - Rs 10000000 (ie INR 20 - 100 LPA)

Min Experience: 3+ years
Location: India
JobType: full-time

We are looking for a highly skilled GPU Compute, MLIR Compiler, and Kernel Optimization Engineer with deep expertise in GPU compute, MLIR-based code generation, and end-to-end performance optimization for AI workloads. In this role, you will design, optimize, and deploy high-performance GPU compute kernels, build and extend MLIR compiler backends, and collaborate closely with ML, runtime, and hardware teams to push the limits of performance on modern GPU architectures.

Requirements

Key Responsibilities

• 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 leve...