Lead Kernel Engineer/Architect

Wir suchen einen Lead Kernel Engineer/Architect, der unser Team in der Schweiz im 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 in großem 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 Werkzeugen
• 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 von klarer Dokumentation, APIs und unterstützenden 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 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 Hochleistungsysteme

• Kenntnisse in der Low-Level-Beschleunigerprogrammierung (CUDA, Triton, Pallas)
• Vertrautheit mit ML-Frameworks wie JAX oder PyTorch und Optimierungstechniken für Aufmerksamkeitslayer, Mixture of Experts (MoE) und Präzisionstuning
• 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

• 5 Wochen Urlaub
• EPAM Employee Stock Purchase Plan (ESPP)
• Erweiterter Elternurlaub
• Erweiterter Pensionsplan
• Tägliche Krankheitszulagenversicherung
• Mitarbeiterunterstützungsprogramm
• Globale Geschäftsreise-Kranken- und Unfallversicherung
• Lern- und Entwicklungsmöglichkeiten, einschließlich interner Schulungen und Coachings, beruflicher Zertifizierungen und Kurse

Bitte beachten Sie, dass alle Angebote einer entsprechenden Hintergrundüberprüfung unterliegen. Wir akzeptieren keine Lebensläufe von Recruiting- oder Personalvermittlungsagenturen. Für diese Position können wir Bewerbungen von folgenden Personen berücksichtigen: Schweizer Staatsangehörige, EU/EFTA-Staatsangehörige, Drittstaatsangehörige mit Wohnsitz in der Schweiz und einem entsprechenden Arbeitsvisum, sowie vertriebene Personen aus der Ukraine, die sich derzeit in der Schweiz befinden und über ein S-Visum verfügen oder bereits beantragt haben.