Lead Kernel Engineer/Architect

Wir suchen einen Lead Kernel Engineer/Architect, der unser Team in der Schweiz im hybriden Arbeitsmodus verstärkt. Sind Sie leidenschaftlich daran interessiert, fortschrittliche Hardware-Beschleuniger bis an ihre Grenzen zu bringen? Schließen Sie sich uns an, um die Zukunft der KI-Leistung und Skalierbarkeit zu gestalten.

In dieser Rolle werden Sie die Optimierung kritischer maschineller Lernoperationen für das Training und die Inferenz in großem Maßstab vorantreiben, indem Sie mit modernster Hardware wie TPUs und GPUs, fortschrittlichen ML-Modellen und Leistungstoolchains arbeiten. Ihre Arbeit ermöglicht schnellere KI-Forschung und Produktionsbereitstellungen auf Cloud-Plattformen und innerhalb von Open-Source-Ökosystemen.

Sie werden 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
• Zusammenarbeit mit Entwicklern von ML-Frameworks (z. B. JAX, PyTorch) und Compiler-Teams (XLA/MLIR), um benutzerdefinierte Kernels zu integrieren und Leistungsengpässe zu reduzieren
• Verfolgen von Fortschritten in der Beschleunigerhardware, Compiler-Technologie und dem Design von KI-Modellen, um Möglichkeiten zur Optimierung auf Kernel-Ebene zu identifizieren
• Entwicklung 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 Aufmerksamkeitslayer, Mixture of Experts (MoE) und Präzisionsanpassung
• Starkes Verständnis moderner Hardware-Beschleuniger, einschließlich Pipeline, Datenbewegung und heterogenem Rechnen
• Kenntnisse der Compiler-Prinzipien 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
• Programm zur Unterstützung von Mitarbeitern
• Globale Reise-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.