GPU Compute & MLIR Engineer

3 weeks ago
Full-time
Mid Level
Software Development
Weekday

Weekday

Weekday helps companies hire engineers who are vouched by other software engineers, enabling passive income for engineers. They offer services like drafting outreach messages, shortlisting candidates, and conducting reference checks. Backed by Y Combin...

Construction & Engineering
11-50
Founded 2020

Description

  • Develop and optimize GPU compute kernels for OpenCL and Vulkan backends to support high-throughput AI/ML workloads.
  • Design, build, and extend MLIR dialects across frontend, graph-level, tensor, runtime, and low-level abstraction layers.
  • Implement MLIR compiler passes and transformations such as tiling, fusion, bufferization, vectorization, and lowering pipelines.
  • Profile compiled kernels using GPU counters and vendor-specific profilers to identify bottlenecks and improve performance.
  • Build and maintain GPU runtime infrastructure, including memory management, pipeline setup, command buffer orchestration, and resource scheduling.
  • Develop code generation pipelines that lower tensor IR through MLIR to efficient OpenCL and Vulkan kernels.
  • Implement performance-critical schedules such as tiling, loop fusion, parallelism, and caching in MLIR-based backends.
  • Collaborate with framework teams to optimize end-to-end model lowering for computer vision and LLM workloads.
  • Design and implement compiler and runtime components using modern C/C++ for high-performance systems.

Requirements

  • 3+ years of experience.
  • Strong hands-on experience with the MLIR framework, including custom dialects, compiler passes, and end-to-end lowering pipelines.
  • Deep expertise in MLIR frontend dialects such as TOSA, StableHLO, or ONNX-MLIR.
  • Experience with MLIR graph-level IR for operation fusion, shape inference, and graph transformations.
  • Experience with MLIR tensor IR using Linalg, Tensor, and Vector dialects, including tiling and fusion strategies.
  • Experience with MLIR runtime and low-level dialects such as Bufferization, MemRef, SCF, GPU, and LLVM.
  • Strong hands-on experience in OpenCL programming, including kernel development, memory model, work-group/work-item optimization, and runtime management.
  • Solid understanding of Vulkan compute programming, including descriptor management, compute pipelines, synchronization primitives, and runtime internals.
  • Strong understanding of GPU architecture, memory hierarchies, and asynchronous compute.
  • Proficiency in C/C++ for system-level development.
  • Experience with kernel profiling and bottleneck analysis on GPU platforms.
  • Strong background in machine learning fundamentals for both Computer Vision and Large Language Model workloads.
  • Must-have skills: GPU computing, MLIR, and C/C++.
  • Good-to-have skills: Vulkan, kernel development, and OpenCL.

Benefits

  • Salary range: Rs 2,000,000 to Rs 10,000,000 per year (INR 20–100 LPA).
  • Full-time position.
  • Location: India.

Interested in this position?

Apply directly on the company website

Apply Now

Similar Roles

Senior Director, Artificial Intelligence

Cellanome 1-10 Biotechnology

Cellanome is hiring a Senior Director of Artificial Intelligence to lead AI/ML strategy and hands-on model development for its computational biology platform.

Deep Learning Machine Learning Python PyTorch TensorFlow
9 hours, 13 minutes ago

Senior Machine Learning Engineer, ML Efficiency

Reddit 1K-5K Internet Software & Services

Reddit is hiring a senior engineer to build an Ads ML Efficiency function that makes model training and inference faster, cheaper, safer, and more scalable.

Machine Learning PyTorch
9 hours, 13 minutes ago

Research Engineer

Talentpluto, Inc. 1-10 Recruiting

Research Engineer at a YC-backed AI training data infrastructure company building automated systems to verify and improve data quality across a decentralized marketplace.

9 hours, 28 minutes ago

Research Engineer

Talentpluto, Inc. 1-10 Recruiting

A YC-backed AI training data infrastructure company is hiring a remote Research Engineer to build systems that verify and improve data quality across its decentralized marketplace.

9 hours, 28 minutes ago

You're on a roll! Sign up now to keep applying.

Sign Up

Already have an account? Log in

Used by 14,729+ remote workers