Lead Kernel Engineer/Architect (m/f/d)

Wir suchen einen Lead Kernel Engineer/Architect, der unser Team in Deutschland in einem hybriden Arbeitsmodus verstärkt. Bist du leidenschaftlich daran interessiert, fortschrittliche Hardware-Beschleuniger bis an ihre Grenzen zu bringen? Schließe dich uns an, um die Zukunft der KI-Leistung und Skalierbarkeit zu gestalten. In dieser Rolle wirst du die Optimierung kritischer maschineller Lernoperationen für das Training und die Inferenz im großen Maßstab vorantreiben, indem du mit modernster Hardware wie TPUs und GPUs, fortschrittlichen ML-Modellen und Leistungstoolchains arbeitest. Deine Arbeit wird schnellere KI-Forschung und Produktionsbereitstellungen auf Cloud-Plattformen und innerhalb von Open-Source-Ökosystemen ermöglichen.

Du wirst mit Forschern, Compiler-Ingenieuren und Framework-Entwicklern zusammenarbeiten, um optimierte, leistungsstarke Lösungen zu liefern, die den Standard für moderne KI-Berechnungen setzen.

• Entwerfen und Optimieren von Hochleistungs-Kernels für TPU- und GPU-Architekturen unter Verwendung von Low-Level-Programmierframeworks wie Pallas, Triton oder Mosaic
• Aufbauen und Pflegen von Leistungsinfrastrukturen, einschließlich Benchmarking-Suiten, Autotuning-Systemen, Regressionstest-Frameworks und Tools
• Zusammenarbeiten mit ML-Framework-Entwicklern (z.B. JAX, PyTorch) und Compiler-Teams (XLA/MLIR), um benutzerdefinierte Kernels zu integrieren und Leistungsengpässe zu reduzieren
• Fortschritte in der Beschleunigerhardware, Compilertechnologie und dem Design von KI-Modellen verfolgen, um Möglichkeiten zur Optimierung auf Kernel-Ebene zu identifizieren
• Entwickeln klarer Dokumentationen, APIs und unterstützender OSS-Komponenten, die die Benutzerfreundlichkeit und Akzeptanz für Entwickler verbessern
• Analysieren und Lösen komplexer Leistungsprobleme, die große verteilte Trainings- und Inferenzsysteme betreffen

• Abschluss in Informatik oder gleichwertige praktische Erfahrung
• 12+ Jahre Berufserfahrung in der Softwareentwicklung oder Systemprogrammierung
• 5+ Jahre Erfahrung in der Softwareentwicklung mit C++ oder Python
• 3+ Jahre Erfahrung in der Testung, Wartung oder Einführung von Softwareprodukten und mindestens 1 Jahr in Softwaredesign oder Architektur
• Praktische Erfahrung in der Leistungsoptimierung auf Kernel-Ebene für Beschleuniger oder Hochleistungssysteme

• Kenntnisse in der Low-Level-Beschleunigerprogrammierung (CUDA, Triton, Pallas)
• Vertrautheit mit ML-Frameworks wie JAX oder PyTorch und Optimierungstechniken für Attention-Schichten, Mixture of Experts (MoE) und Präzisionsanpassung
• Starkes Verständnis moderner Hardware-Beschleuniger, einschließlich Pipeline, Datenbewegung und heterogenem Rechnen
• Kenntnisse der Compilerprinzipien und Zwischenrepräsentationen (z.B. MLIR, OpenXLA)
• Erfahrung im Aufbau von OSS-Entwicklerinfrastrukturen, APIs und leistungsrelevanten Bibliotheken
• Ausgezeichnete Problemlösungsfähigkeiten und die Fähigkeit zur Zusammenarbeit in funktionsübergreifenden Ingenieureumgebungen

• 30 Tage Urlaub pro Jahr
• Betriebliche Altersvorsorge
• Regelmäßige Leistungsbeurteilungen
• Rabatt auf Fitness-First Black Membership
• bitkom - Corporate Benefits
• Employee Stock Purchase Plan (ESPP) (unterliegt bestimmten Berechtigungsanforderungen)
• Lern- und Entwicklungsmöglichkeiten, einschließlich interner Schulungen und Coachings, beruflicher Zertifizierungen und Kurse
• Freundliches und angenehmes Team
• Regelmäßige Unternehmens- und soziale Veranstaltungen
• Flexible und remote Arbeitsmöglichkeiten
• Ausgezeichneter Arbeitsplatz: Great Place To Work zertifiziert 2026, Kununu (Top Company 2022-2026), NewWork Business Award 2025 für herausragende Kultur, Innovation und Mitarbeiterzufriedenheit.