Lead Kernel Engineer/Architect

Wir suchen einen Lead Kernel Engineer/Architect, der unserem Team in der Schweiz in einem hybriden Arbeitsmodus beitritt. 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 im großen Maßstab vorantreiben und mit modernster Hardware wie TPUs und GPUs sowie fortschrittlichen ML-Modellen und Leistungstoolchains arbeiten. Ihre Arbeit ermöglicht schnellere KI-Forschung und Produktionsbereitstellungen auf Cloud-Plattformen und innerhalb von Open-Source-Ökosystemen.

Zu Ihren Aufgaben gehören:

• Entwurf und Optimierung von Hochleistungs-Kernels für TPU- und GPU-Architekturen unter Verwendung von Low-Level-Programmierframeworks wie Pallas, Triton oder Mosaic
• Aufbau und Wartung 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
• Verfolgung von Fortschritten in der Beschleunigerhardware, Compilertechnologie und dem Design von KI-Modellen zur Identifizierung von Möglichkeiten zur Optimierung auf Kernel-Ebene
• Entwicklung klarer Dokumentationen, APIs und unterstützender OSS-Komponenten, die die Benutzerfreundlichkeit und Akzeptanz für Entwickler verbessern
• Analyse und Lösung komplexer Leistungsprobleme, die große verteilte Trainings- und Inferenzsysteme betreffen

Voraussetzungen:

• 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 Hochleistungssysteme

Wünschenswert:

• 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

Wir bieten:

• 5 Wochen Urlaub
• EPAM Employee Stock Purchase Plan (ESPP)
• Erweiterter Elternurlaub
• Erweiterte Altersvorsorge
• Tägliche Krankheitszulagenversicherung
• Programm zur Unterstützung von Mitarbeitern
• 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 Personalvermittlungs- oder Zeitarbeitsagenturen. Für diese Position können wir Bewerbungen von folgenden Personen in Betracht ziehen: Schweizer Staatsangehörige, EU/EFTA-Staatsangehörige, Drittstaatsangehörige mit Wohnsitz in der Schweiz und mit einer entsprechenden Arbeitserlaubnis, sowie vertriebene Personen aus der Ukraine, die sich derzeit in der Schweiz befinden und über ein S-Visum verfügen oder bereits beantragt haben.