12 Oct
|
Oxmiq Labs
|
Hyderabad
12 Oct
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 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.
• Solid 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 fundamen
📌 Compiler Engineer (Hyderabad)
🏢 Oxmiq Labs
📍 Hyderabad