Senior/Staff AI Graph Compiler Engineer

Axelera AI ist kein gewöhnliches Deep-Tech-Startup. Wir schaffen die nächste Generation der KI-Plattform, um jedem zu helfen, der die Menschheit voranbringen und die Welt um uns herum verbessern möchte. In nur vier Jahren haben wir insgesamt 370 Millionen Dollar gesammelt und ein erstklassiges Team von über 220 Mitarbeitern (darunter mehr als 49 PhDs mit über 40.000 Zitierungen) aufgebaut, sowohl remote aus 18 verschiedenen Ländern als auch mit Büros in Belgien, Frankreich, der Schweiz, Italien, dem Vereinigten Königreich und mit Hauptsitz im High Tech Campus in Eindhoven, Niederlande. Wir haben auch unsere Metis™ KI-Plattform gestartet, die eine 3-5-fache Steigerung der Effizienz und Leistung erreicht und einen starken Geschäftspipeline von über 100 Millionen Dollar aufweist. Unser unerschütterliches Engagement für Innovation hat uns fest als globalen Branchenpionier etabliert. Bist du bereit für die Herausforderung?

Wir suchen einen Frontend / Graph-Level Compiler Engineer, der unser wachsendes Compiler-Team bei Axelera AI verstärkt. In dieser Rolle spielst du eine Schlüsselrolle bei der Entwicklung und Optimierung unseres MLIR-basierten Compiler-Stacks, der eine effiziente Ausführung von KI-Workloads auf modernsten heterogenen Hardware-Architekturen ermöglicht. Du wirst eng mit KI-Forschern, Compiler-Ingenieuren und Hardware-Architekten zusammenarbeiten und mit einem talentierten Team von Ingenieuren in ganz Europa kooperieren. Dies ist deine Chance, an modernen KI-Beschleunigungsarchitekturen zu arbeiten, Compiler-Technologie voranzutreiben und einen echten Einfluss in einem schnelllebigen Startup-Umfeld zu haben.

Hauptverantwortlichkeiten

• Entwerfen, Implementieren und Warten von Frontend- und Graph-Level-Compiler-Komponenten unter Verwendung von MLIR
• Entwickeln und Optimieren von Graph-Level-Transformationen wie Operatorfusion, Konstantenfaltung, Operatorabsenkung, Graphpartitionierung und anderen leistungsrelevanten Optimierungen
• Erweitern und Warten von MLIR-Dialekten, -Pässen und -Infrastrukturen zur Unterstützung von KI-Workloads
• Integrieren und Senken von KI-Modellen aus Frameworks wie PyTorch, ONNX und TensorFlow in interne Compiler-Darstellungen
• Zusammenarbeiten mit Hardware- und Backend-Compiler-Teams, um eine effiziente Zuordnung von KI-Workloads zu heterogenen Architekturen sicherzustellen
• Unterstützen und Mentoren von Teammitgliedern bei der Annahme und effektiven Nutzung der MLIR-Infrastruktur
• Analysieren von Modellgraphen und Implementieren von Optimierungen zur Verbesserung der Leistung, Speichereffizienz und Ausführungseffizienz
• Beitragen zum Design und zur Weiterentwicklung der gesamten Compiler-Architektur und -Werkzeuge
• Debuggen, Profilieren und Verbessern der Compiler-Leistung und -Korrektheit

Qualifikationen

• Master oder PhD in Informatik oder einem verwandten technischen Bereich
• 3-5 Jahre Erfahrung in einer Software-Engineering-Rolle, davon idealerweise 2 Jahre Erfahrung mit Deep-Learning-Frameworks und KI-Systemen
• Erfahrung mit MLIR, einschließlich Dialekten, Pässen und Compiler-Infrastruktur
• Solides Verständnis der Prinzipien des Compiler-Designs und der Zwischenrepräsentationen, insbesondere graphbasierter IRs
• Erfahrung mit KI-Modell-Frameworks wie PyTorch (bevorzugt), ONNX oder TensorFlow
• Nachweisliche Erfahrung in der Implementierung von Graph-Level-Optimierungen wie Operatorfusion, Konstantenfaltung, Graphpartitionierung oder ähnlichen Transformationen
• Starke Programmierkenntnisse in C++ und Python
• Erfahrung in kollaborativen Ingenieurteams
• Ausgezeichnete Kommunikationsfähigkeiten und die Bereitschaft, Wissen zu teilen und andere zu betreuen

Wünschenswert

• Erfahrung mit benutzerdefinierten KI-Beschleunigern oder spezialisierter Hardware
• Hintergrund in Computerarchitektur, insbesondere heterogenen Systemen (z.B. CPU + NPU, GPU oder dedizierte Beschleuniger)
• Erfahrung mit KI-Compiler-Stacks wie Torch-MLIR, TVM, XLA, Glow oder ähnlichem
• Erfahrung in der Optimierung von KI-Workloads hinsichtlich Leistung und Effizienz
• Vertrautheit mit Frontend-Modellaufnahme, Graphsenkung und Compiler-Pipelines

Standort

Wir bieten eine flexible Arbeitsgestaltung mit Optionen, um: Von einem unserer Axelera AI-Büros (Leuven in Belgien, Amsterdam und Eindhoven in den Niederlanden, Zürich in der Schweiz, Florenz und Mailand in Italien oder Bristol im Vereinigten Königreich) zu arbeiten, wenn du bereits in der Nähe bist. Vollständig remote aus jedem europäischen Land (einschließlich des Vereinigten Königreichs), in dem du dich bereits befindest, zu arbeiten. Mit uns umzuziehen und von Italien (Florenz oder Mailand) oder den Niederlanden (Amsterdam oder Eindhoven) aus zu arbeiten. Bitte beachte, dass Kandidaten, die daran interessiert sind, in Belgien oder Italien ansässig zu sein, Vorrang haben.

Was wir bieten

Dies ist deine Chance, Teil einer dynamischen, schnell wachsenden, internationalen Organisation zu werden. Wir bieten ein attraktives Vergütungspaket, einschließlich eines Rentenplans, umfangreicher Mitarbeiterversicherungen und der Möglichkeit, Unternehmensanteile zu erhalten. Eine offene Kultur, die Kreativität und kontinuierliche Innovation unterstützt, erwartet dich. Kollaborativer Besitz und Freiheit mit Verantwortung sind charakteristisch für die Art und Weise, wie wir als Team handeln und arbeiten. Bei Axelera AI setzen wir uns uneingeschränkt für Chancengleichheit ein und legen größten Wert auf Vielfalt. Unser fester Einsatz gilt der Schaffung eines warmen und integrativen Umfelds, das jedes Mitglied unseres Teams ermächtigt und feiert. Wir heißen Bewerber aus allen Hintergründen willkommen, um gemeinsam die Zukunft der KI zu gestalten.