Switzerland Job Openings

Axelera AI

Staff AI Compiler Engineer

Zürich

FULL TIME

September 5, 2024

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+ Ph Ds 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 Ph D 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 Py Torch, 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.
New Job Alerts
AFRY

Projektingenieur:in Tragwerksplanung Ingenieurbau 80-100% - Zürich

Zürich

November 20, 2024

View Job Description
AFRY

BIM/VDC Projektingenieur:in Infrastrukturbau 80-100% - Zürich

Zürich

November 20, 2024

View Job Description
AFRY

Capo progetto opere sotterranee - Tunnelling 80-100% (f/m/d) - dalla nuova sede in Bellinzona o da Zurigo

Bellinzona

November 20, 2024

View Job Description
AFRY

Leitende:r Projektingenieur:in Tiefbau/Bahnbau 80-100% - Zürich

Zürich

November 20, 2024

View Job Description
Franke

Supply Chain Manager Switzerland (w/m) 100%

Aarburg

FULL TIME

November 20, 2024

View Job Description
AFRY

Projektingenieur:in Untertagbau 80-100% - Zürich

Zürich

November 20, 2024

View Job Description
AFRY

Zeichner:in / BIM Modellierer:in / Konstrukteur:in Tragwerksplanung Ingenieurbau 80-100% - Zürich

Zürich

November 20, 2024

View Job Description
AFRY

Praktikant:in Tragwerksplanung Ingenieurbau 100% - Zürich

Zürich

FULL TIME

November 20, 2024

View Job Description
AFRY

Projektingenieur:in Infrastrukturbau 80-100% - Zürich

Zürich

November 20, 2024

View Job Description
AFRY

PROJEKTLEITER:IN BAUPHYSIK - ZÜRICH

Zürich

November 20, 2024

View Job Description
Looking for similar job?
Agroscope is looking for a:

Staff Research Scientist - Biostatistics and Quantitative Genetics

Wädenswil

PART TIME

August 18, 2024

View Job Description
SIXT Switzerland

Fahrzeugaufbereiter - Car Support Staff (m/w/d) 100%

Basel

FULL TIME

August 29, 2024

View Job Description
SIXT Switzerland

Fahrzeugaufbereiter - Car Support Staff (m/w/d) 100%

Zürich

FULL TIME

August 30, 2024

View Job Description
International Organization for Migration

Intern - Global Staff Association Committee (GSAC)

Geneva

August 22, 2024

View Job Description
Viamonde SA

Domestic Staff

August 21, 2024

View Job Description
Viamonde SA

Domestic Staff & Kitchen Assistant

August 7, 2024

View Job Description
See What’s New: Axelera AI Job Opportunities
Axelera AI

Staff AI Compiler Engineer

Zürich

FULL TIME

September 5, 2024

View Job Description
Axelera AI

Build Engineer

Zürich

FULL TIME

August 30, 2024

View Job Description
View More Jobs by Axelera AI
New Job Alerts
AFRY

Projektingenieur:in Tragwerksplanung Ingenieurbau 80-100% - Zürich

Zürich

November 20, 2024

View Job Description
AFRY

BIM/VDC Projektingenieur:in Infrastrukturbau 80-100% - Zürich

Zürich

November 20, 2024

View Job Description
AFRY

Capo progetto opere sotterranee - Tunnelling 80-100% (f/m/d) - dalla nuova sede in Bellinzona o da Zurigo

Bellinzona

November 20, 2024

View Job Description
AFRY

Leitende:r Projektingenieur:in Tiefbau/Bahnbau 80-100% - Zürich

Zürich

November 20, 2024

View Job Description
Franke

Supply Chain Manager Switzerland (w/m) 100%

Aarburg

FULL TIME

November 20, 2024

View Job Description
AFRY

Projektingenieur:in Untertagbau 80-100% - Zürich

Zürich

November 20, 2024

View Job Description
AFRY

Zeichner:in / BIM Modellierer:in / Konstrukteur:in Tragwerksplanung Ingenieurbau 80-100% - Zürich

Zürich

November 20, 2024

View Job Description
AFRY

Praktikant:in Tragwerksplanung Ingenieurbau 100% - Zürich

Zürich

FULL TIME

November 20, 2024

View Job Description
AFRY

Projektingenieur:in Infrastrukturbau 80-100% - Zürich

Zürich

November 20, 2024

View Job Description
AFRY

PROJEKTLEITER:IN BAUPHYSIK - ZÜRICH

Zürich

November 20, 2024

View Job Description