AI Compiler Engineer
--10xEngineers--
Compiler Engineer – AI & ML Compilers
Location: Lahore
Employment Type: Full-time
About the Role
We are looking for a Compiler Engineer for the AI Compiler team at 10xE. The candidate will work on MLIR-based AI compilers, contributing to performance optimizations, backend code generation, and ML model deployment.
The ideal candidate should have strong C++ skills and a solid understanding of computer architecture and operating systems. Open-source contributions to LLVM, MLIR, LLama.cpp, or TFLite or related frameworks are highly preferred. Prior experience with RISC-V ISA and RISC-V Vector ISA is a plus. The candidate should also be comfortable working with Assembly language and low-level optimizations.
Key Responsibilities
Develop and optimize AI/ML compilers for custom hardware.
Implement and improve compiler passes, IR transformations, and the compiler backend.
Work on RISC-V ISA and RISC-V Vector ISA support in AI compilers.
Debug and resolve performance bottlenecks in generated code.
Collaborate with ML engineers and hardware teams to optimize compiler-generated code for AI workloads.
Required Skills
Experience with C++ (modern C++ preferred, e.g., C++11/14/17)
Solid understanding of computer architecture and operating systems
Experience with Assembly language and low-level programming
Awareness of basic machine learning and deep learning algorithms, including CNNs and Transformer architecture
Preferred Qualifications
Experience contributing to LLVM, MLIR, or AI compilers like TVM, IREE, etc.
Knowledge of RISC-V ISA and RISC-V Vector ISA.
Experience with parallel computing, SIMD optimizations, vectorization techniques, and hardware-aware optimizations.
Experience with QEMU for system emulation.
Experience working with large open-source C++ codebases such as LLVM, MLIR, TensorFlow, PyTorch, Chromium, or similar.
Why join us?
Contribute to AI compiler projects, solving real-world performance and deployment challenges.
Your work will directly influence major open-source projects like LLVM and MLIR.
Gain expertise in compilers, RISC-V, and ML acceleration.
Flexible work environment
If you're passionate about AI compilers and high-performance code generation, we’d love to hear from you.