06 Sep
|
Oxmiq Labs
|
India
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 efficient 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 contemporary 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 funda
📌 Compiler Engineer (India)
🏢 Oxmiq Labs
📍 India