Senior/Staff AI Graph Compiler Engineer
UK Visa Sponsorship Analytics
Analytics are greyed out due to low classification confidence (35.0%).
Occupation Type
Computer system and equipment installers and servicers
Occupation Code Skill LevelMedium Skilled
Sponsorship Salary Threshold
£41,700
(£21.38 per hour)
Standard minimum applies
Above analytics are generated algorithmically based on job titles and may not always be the same as the company's job classification. You can also check detailed occupation eligibility, and salary criteria on our UK Visa Eligible Occupations & Salary Thresholds page.
Disclaimer: Hunt UK Visa Sponsors aggregates job listings from publicly available sources, such as search engines, to assist with your job hunting. We do not claim affiliation with Axelera AI. For the most up-to-date job details, please visit the official website by clicking "Apply Now."
Description
About Us
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 advancing humanity and improve the world around us.
In just four years, we have raised a total of $370 million and have built a world-class team of 220+ employees (including 49+ PhDs with more than 40,000 citations), both remotely from 18 different countries and with offices in Belgium, France, Switzerland, Italy, the UK, headquartered 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 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
We are looking for a Frontend / Graph-level Compiler Engineer to join our growing compiler team at Axelera AI. In this role, you will play a key part in developing and optimizing our MLIR-based compiler stack, enabling efficient execution of AI workloads on cutting-edge heterogeneous hardware architectures.
You will work closely with AI researchers, compiler engineers, and hardware architects, collaborating with a talented team of engineers across Europe. This is your chance to work on cutting-edge AI acceleration architectures, advance compiler technology, and make a real impact in a fast-moving startup environment.
Key responsibilities:
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 advancing humanity and improve the world around us.
In just four years, we have raised a total of $370 million and have built a world-class team of 220+ employees (including 49+ PhDs with more than 40,000 citations), both remotely from 18 different countries and with offices in Belgium, France, Switzerland, Italy, the UK, headquartered 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 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
We are looking for a Frontend / Graph-level Compiler Engineer to join our growing compiler team at Axelera AI. In this role, you will play a key part in developing and optimizing our MLIR-based compiler stack, enabling efficient execution of AI workloads on cutting-edge heterogeneous hardware architectures.
You will work closely with AI researchers, compiler engineers, and hardware architects, collaborating with a talented team of engineers across Europe. This is your chance to work on cutting-edge AI acceleration architectures, advance compiler technology, and make a real impact in a fast-moving startup environment.
Key responsibilities:
- 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
- Master’s or PhD in Computer Science or a related technical field
- 3-5 years of experience in a Software Engineering role, thereof ideally 2 years of experience with Deep Learning frameworks and AI systems
- Experience working with custom AI accelerators or specialized hardware
- Background in computer architecture, especially heterogeneous systems (e.g., CPU + NPU, GPU, or dedicated accelerators)
- Experience with AI compiler stacks such as Torch-MLIR, TVM, XLA, Glow, or similar
- Experience optimizing AI workloads for performance and efficiency
- Familiarity with frontend model ingestion, graph lowering, and compiler pipelines
We offer a flexible working arrangement, with options to:
What we offer
This is your chance to shape and be part of a dynamic, fast-growing, international organization. We offer an attractive compensation package, including a pension plan, extensive employee insurances and the option to get company shares.
An open culture that supports creativity and continual innovation is awaiting you. Collaborative ownership and freedom with responsibility is characteristic for the way we act and work as a team.
At Axelera AI, we wholeheartedly embrace equal opportunity and hold diversity in the highest regard. Our steadfast commitment is to cultivate a warm and inclusive environment that empowers and celebrates every member of our team. We welcome applicants from all backgrounds to join us in shaping the future of AI.
- Work from one of our Axelera AI offices (Leuven in Belgium, Amsterdam and Eindhoven in the Netherlands, Zurich in Switzerland, Florence and Milan in Italy or Bristol in the United Kingdom) if you're already based in the vicinity.
- Work fully remotely from any European country (incl. the UK) you are already in.
- Relocate with us and work from Italy (Florence or Milan) or the Netherlands (Amsterdam or Eindhoven).
What we offer
This is your chance to shape and be part of a dynamic, fast-growing, international organization. We offer an attractive compensation package, including a pension plan, extensive employee insurances and the option to get company shares.
An open culture that supports creativity and continual innovation is awaiting you. Collaborative ownership and freedom with responsibility is characteristic for the way we act and work as a team.
At Axelera AI, we wholeheartedly embrace equal opportunity and hold diversity in the highest regard. Our steadfast commitment is to cultivate a warm and inclusive environment that empowers and celebrates every member of our team. We welcome applicants from all backgrounds to join us in shaping the future of AI.