Match score not available

Staff AI Compiler Engineer

Remote: 
Full Remote
Contract: 
Experience: 
Senior (5-10 years)
Work from: 
Texas (USA), United States

Offer summary

Qualifications:

Master’s or PhD in CS/Electrical Engineering., 5+ years of software engineering experience., Strong background in AI compiler development., Familiarity with C++, Python, PyTorch., Knowledge of TVM, MLIR, Glow frameworks..

Key responsabilities:

  • Lead design and development of Neural Network compiler.
  • Innovate with compiler optimizations for In-Memory Computing.
  • Guide complex projects to support various workloads.
  • Collaborate for technical alignment and stack optimization.
  • Mentor, model excellence, and drive technical solutions.
Axelera AI logo
Axelera AI Scaleup https://axelera.ai/
51 - 200 Employees
See more Axelera AI offers

Job description

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. 

Key Responsibilities: 

  • 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.

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. 

Location 

We offer a flexible working arrangement, with options to: 

  • 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). 

  • Work fully remotely from any European country (including the UK) you are currently based in. 

  • Relocate and work from Amsterdam or Eindhoven (Netherlands) or Florence (Italy). 

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 receive company shares. 

An open culture that supports creativity and continual innovation is awaiting you. Collaborative ownership and freedom with responsibility are central to how we 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. 

Required profile

Experience

Level of experience: Senior (5-10 years)
Industry :
Spoken language(s):
English
Check out the description to know which languages are mandatory.

Other Skills

  • Verbal Communication Skills
  • Scheduling
  • Leadership
  • Mentorship
  • Collaboration
  • Problem Solving

Artificial Intelligence Engineer Related jobs