Switzerland Job Openings
Axelera AI
Staff AI Compiler Engineer
Zürich
FULL TIME
September 5, 2024
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 DescriptionAFRY
BIM/VDC Projektingenieur:in Infrastrukturbau 80-100% - Zürich
Zürich
November 20, 2024
View Job DescriptionAFRY
Capo progetto opere sotterranee - Tunnelling 80-100% (f/m/d) - dalla nuova sede in Bellinzona o da Zurigo
Bellinzona
November 20, 2024
View Job DescriptionAFRY
Leitende:r Projektingenieur:in Tiefbau/Bahnbau 80-100% - Zürich
Zürich
November 20, 2024
View Job DescriptionFranke
Supply Chain Manager Switzerland (w/m) 100%
Aarburg
FULL TIME
November 20, 2024
View Job DescriptionAFRY
Zeichner:in / BIM Modellierer:in / Konstrukteur:in Tragwerksplanung Ingenieurbau 80-100% - Zürich
Zürich
November 20, 2024
View Job DescriptionAFRY
Praktikant:in Tragwerksplanung Ingenieurbau 100% - Zürich
Zürich
FULL TIME
November 20, 2024
View Job DescriptionAFRY
Projektingenieur:in Infrastrukturbau 80-100% - Zürich
Zürich
November 20, 2024
View Job DescriptionLooking for similar job?
Agroscope is looking for a:
Staff Research Scientist - Biostatistics and Quantitative Genetics
Wädenswil
PART TIME
August 18, 2024
View Job DescriptionSIXT Switzerland
Fahrzeugaufbereiter - Car Support Staff (m/w/d) 100%
Basel
FULL TIME
August 29, 2024
View Job DescriptionSIXT Switzerland
Fahrzeugaufbereiter - Car Support Staff (m/w/d) 100%
Zürich
FULL TIME
August 30, 2024
View Job DescriptionInternational Organization for Migration
Intern - Global Staff Association Committee (GSAC)
Geneva
August 22, 2024
View Job DescriptionNew Job Alerts
AFRY
Projektingenieur:in Tragwerksplanung Ingenieurbau 80-100% - Zürich
Zürich
November 20, 2024
View Job DescriptionAFRY
BIM/VDC Projektingenieur:in Infrastrukturbau 80-100% - Zürich
Zürich
November 20, 2024
View Job DescriptionAFRY
Capo progetto opere sotterranee - Tunnelling 80-100% (f/m/d) - dalla nuova sede in Bellinzona o da Zurigo
Bellinzona
November 20, 2024
View Job DescriptionAFRY
Leitende:r Projektingenieur:in Tiefbau/Bahnbau 80-100% - Zürich
Zürich
November 20, 2024
View Job DescriptionFranke
Supply Chain Manager Switzerland (w/m) 100%
Aarburg
FULL TIME
November 20, 2024
View Job DescriptionAFRY
Zeichner:in / BIM Modellierer:in / Konstrukteur:in Tragwerksplanung Ingenieurbau 80-100% - Zürich
Zürich
November 20, 2024
View Job DescriptionAFRY
Praktikant:in Tragwerksplanung Ingenieurbau 100% - Zürich
Zürich
FULL TIME
November 20, 2024
View Job DescriptionAFRY
Projektingenieur:in Infrastrukturbau 80-100% - Zürich
Zürich
November 20, 2024
View Job Description