RemoteLeaf Logo
Log in Sign up

GPU Compute & MLIR Engineer

14 hours, 22 minutes ago
India
Full-time
Mid Level
Machine Learning Engineer
Software Development
C C++ Computer Vision LLM Vulkan
Apply Now
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
View All Jobs 119

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

Epoch AI

Researcher, Benchmark Reviews

Epoch AI 1-10 Professional Services

Epoch AI is hiring a fully remote Researcher to critically analyze AI benchmarks and publish public-facing reviews that clarify what benchmark performance does and does not indicate about AI capabilities.

Anywhere Full-time Mid Level AI (Artificial Intelligence) Research Scientist
$100k-$200k
Machine Learning
14 hours, 37 minutes ago
Nacre Capital

Principal / Staff Software Engineer - Backend, MLOps and Cloud Infrastructure

Nacre Capital 11-50 Capital Markets

Seed-X is hiring a Principal / Staff Software Engineer to lead backend, MLOps, and cloud platform architecture for its AI-driven AgTech systems that analyze seed quality at scale.

Portugal Romania Bulgaria Spain Hungary Full-time Lead Backend Engineer Machine Learning Engineer
AWS CI/CD Datadog DNS Docker Grafana Kubernetes Microservices MLOps OpenSearch Prometheus Python SQL
1 day, 13 hours ago
Pavago

Full-Stack AI Engineer

Pavago IT Services

A remote Full-Stack AI Engineer role for a client building production-ready AI applications that combine software engineering, machine learning, and cloud infrastructure.

United States Portugal Full-time Mid Level Full-stack Engineer Machine Learning Engineer
Apache Airflow AWS Azure CI/CD Dagster Docker FastAPI Flask GCP HIPAA Hugging Face JavaScript Kubeflow Kubernetes Microservices MLflow Next.js Node.js Prefect Prometheus Python PyTorch React REST API SageMaker Serverless Snowflake SQL TensorFlow TypeScript Vertex AI Vue.js
1 day, 14 hours ago
Xenon7

Senior Data Scientist - Europe

Xenon7 Internet Software & Services

Xenon7 is hiring a Senior Data Scientist for a remote Europe-based contract to build a document intelligence and discovery platform that turns scanned chemical PDFs into searchable, actionable knowledge bases.

Poland Serbia Croatia Spain France Germany Contract Senior Data Scientist Machine Learning Engineer
AWS Computer Vision Deep Learning Generative AI LLM Machine Learning Python PyTorch Scikit-learn TensorFlow
1 day, 14 hours 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