Senior/Staff AI Graph Compiler Engineer

Axelera AIAI Platform

Europe (incl. UK)Full-TimeSenior

Salary not disclosed

Apply NowOpens the employer's application page

Job Details

Master’s or PhD in Computer Science or a related technical field
3-5 years of experience in a Software Engineering role
Ideally 2 years of experience with Deep Learning frameworks and AI systems
Experience with MLIR, including dialects, passes, and compiler infrastructure
Solid understanding of compiler design principles and intermediate representations, especially graph-based Irs
Experience working with AI model frameworks such as PyTorch (preferred), ONNX, or TensorFlow
Proven experience implementing graph-level optimizations such as operator fusion, constant folding, graph partitioning, or similar transformations
Strong programming skills in C++
Strong programming skills in Python
Experience working in collaborative engineering teams
Excellent communication skills and willingness to share knowledge and mentor others

Design, implement, and maintain frontend and graph-level compiler components using MLIR
Develop and optimize graph-level transformations such as operator fusion, constant folding, operator sinking, graph partitioning, and other performance-critical optimizations
Extend and maintain MLIR dialects, passes, and infrastructure to support AI workloads
Integrate and lower AI models from frameworks such as PyTorch, ONNX, and TensorFlow into internal compiler representations
Collaborate with hardware and backend compiler teams to ensure efficient mapping of AI workloads to heterogeneous architectures
Support and mentor team members in adopting and effectively using MLIR infrastructure
Analyze model graphs and implement optimizations to improve performance, memory efficiency, and execution efficiency
Contribute to the design and evolution of the overall compiler architecture and tooling
Debug, profile, and improve compiler performance and correctness

View Full Description & ApplyYou'll be redirected to the employer's site