Staff AI Compiler Engineer

at  Axelera AI

ABOUT US

Axelera AI is not your regular deep-tech startup. We are creating the next-generation AI platform to support anyone who wants to help advance humanity and improve the world around us.
In just three years, we have raised a total of $120 million and built a world-class team of 180+ employees (including 55+ PhDs with more than 40,000 citations), working remotely from 11 different countries and from offices in Belgium, Switzerland, Italy, the UK, and our headquarters at the High Tech Campus in Eindhoven, Netherlands.
We have also launched our Metis AI Platform, which achieves a 3-5x increase in efficiency and performance, and we have visibility into a strong business pipeline exceeding $100 million.
Our unwavering commitment to innovation has firmly established us as a global industry pioneer.
Are you up for the challenge?

POSITION OVERVIEW

As a Staff AI Compiler Engineer, you will be at the forefront of Axelera AI’s Neural Network compiler development. You’ll be responsible for driving the technical direction, solving complex, open-ended problems, and setting priorities for high-impact projects. Your work will shape Axelera AI’s software systems and extend beyond your immediate team, with a focus on delivering scalable, flexible solutions to advance our AI platform.

QUALIFICATIONS:

• Master’s or PhD in Computer Science, Electrical Engineering, or a related technical field.
• 5+ years of experience in software engineering, with at least 2 years focused on Deep Learning and AI systems.
• Strong experience in AI compiler development, e.g. in some of these topics: Workload partitioning for heterogeneous platforms, scheduling and tiling optimizations, managing a multi-level memory hierarchy, synchronization.
• Background in computer architecture with related knowledge of heterogeneous compute systems.
• Familiarity with open-source compiler frameworks such as TVM, MLIR, or Glow.
• Experience in neural network quantization techniques is a plus.
• Proficiency in PyTorch, with a good understanding of deep learning frameworks is a plus.
• Excellent communication and leadership skills, with a high sense of ownership and the ability to drive technical solutions that impact the broader organization.
• Proven to be a team player favoring collaborative decision making over building silos.
• Proficiency in C++ and Python.

• Lead the design, architecture, and hands-on development of Axelera AI’s Neural Network compiler and SDK.
• Drive innovation by designing compiler optimizations, focusing on hardware-aware transformations for In-Memory Computing.
• Execute and technically guide complex projects to ensure our compiler supports workloads beyond classical CNNs (e.g. generative AI) and upcoming generations of our heterogeneous compute platform.
• Collaborate with cross-functional teams to ensure alignment on technical strategy, optimizing the Axelera software stack for scalability and performance.
• Oversee holistically our compilation pipeline and proactively propose changes to ensure best-in-class functional coverage, performance and accuracy.
• Ensure best practices in the team, especially regarding testing methodologies for compilers.
• Mentor and guide junior and senior engineers, modeling a standard of excellence and leadership within the team, thereby being a valuable partner to your manager and peers.