01 Sep
|
Oxmiq Labs
|
Hyderabad
01 Sep
Oxmiq Labs
Hyderabad
Compiler Engineer (GPU Backend)
Oxmiq Labs | Hyderabad, India (On-site)
Experience: 5+ Years
Key Responsibilities
Design and implement compiler infrastructure for GPU-accelerated workloads, including lowering pipelines, optimization passes, and code generation backends.
Develop and optimize compilation passes targeting Oxmiq hardware IP and GPU architectures.
Design and implement segmentation and scheduling strategies for the Oxmiq hardware backend to maximize hardware utilization, throughput, and execution efficiency.
Develop compiler optimizations for operator fusion, tiling, memory planning, and workload partitioning across GPU hardware.
Collaborate with architecture and hardware teams to translate performance requirements into productive compiler transformations and code generation strategies.
Work with kernel engineers to support Triton-based programming models, custom operator lowering, and integration with modern LLM serving frameworks such as vLLM and SGLang.
Optimize compiler output for throughput, latency, memory efficiency, and scalability across AI workloads.
Participate in architecture discussions, design reviews, and code reviews.
Support compiler performance profiling, benchmarking, debugging, and root cause analysis.
Required Qualifications
5+ years of experience in compiler engineering, with a focus on GPU, NPU, or AI accelerator software stacks.
Strong proficiency in C++.
Hands-on experience with at least one modern compiler framework such as MLIR, LLVM, TVM, or IREE.
Experience designing or implementing compiler optimizations, scheduling algorithms, segmentation/partitioning strategies, or lowering pipelines for GPU/NPU hardware.
Solid understanding of compiler concepts including IR transformations, code generation, instruction scheduling, register allocation, vectorization, and memory hierarchy optimization.
Familiarity with GPU programming models such as CUDA or Triton.
Understanding of GPU architecture fundamentals including thread execution
📌 Compiler Engineer (Hyderabad)
🏢 Oxmiq Labs
📍 Hyderabad