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

EPAM Systems
München

We're looking for a Lead Kernel Engineer/Architect to join our team in Germany in a hybrid working mode.

Are you passionate about pushing advanced hardware accelerators to their limits? Join us in shaping the future of AI performance and scalability.

As a Lead Kernel Engineer/Architect, you will drive the optimization of critical machine learning operations for large-scale training and inference, working with cutting-edge hardware like TPUs and GPUs, advanced ML models and performance toolchains. Your work will enable faster AI research and production deployments on cloud platforms and within open-source ecosystems.

In this role, you will collaborate with researchers, compiler engineers and framework developers to deliver optimized, high-performance solutions that set the standard for modern AI computation.

Design and optimize high-performance kernels for TPU and GPU architectures using low-level programming frameworks such as Pallas, Triton or Mosaic Build and maintain performance infrastructure, including benchmarking suites, autotuning systems, regression testing frameworks and tooling Collaborate with ML framework developers (e.g., JAX, PyTorch) and compiler teams (XLA/MLIR) to integrate custom kernels and reduce performance bottlenecks Track advancements in accelerator hardware, compiler technology and AI model design to identify opportunities for kernel-level optimization Develop clear documentation, APIs and supporting OSS components that improve developer usability and adoption Analyze and resolve complex performance issues impacting large-scale distributed training and inference systems Bachelor’s degree or equivalent practical experience 12+ years of industry experience in software engineering or systems programming 5+ years of experience in software development using C++ or Python 3+ years of experience in testing, maintaining or launching software products and at least 1 year in software design or architecture Hands-on experience in performance optimization at the kernel level for accelerators or high-performance systems Proficiency in low-level accelerator programming (CUDA, Triton, Pallas) Familiarity with ML frameworks such as JAX or PyTorch and optimization techniques for attention layers, Mixture of Experts (MoE) and precision tuning Strong understanding of modern hardware accelerators, including pipelining, data movement and heterogeneous compute Knowledge of compiler principles and intermediate representations (e.g., MLIR, OpenXLA) Experience building OSS developer infrastructure, APIs and performance-critical libraries Excellent problem-solving skills and ability to collaborate in cross-functional engineering environments
Veröffentlicht am 2026-06-05

Empfohlene Jobs

Medizinische Fachangestellte (m/w/d)

ARBEITSWELT Personaldienstleistungen GmbH & Co. KG
München

Für unseren Kunden in der Arbeitsmedizin suchen wir ab sofort eine Medizinische Fachangestellte (m/w/d) in München in Festanstellung. Übertarifliche Bezahlung. Ab 19,00 € pro Stunde Vollzeit o…

Details Anzeigen
Veröffentlicht am 2026-08-28

Mitarbeiter Vertriebsinnendienst Gebrauchtwagen (m/w/d)

München

Sie begeistern sich für Vertrieb, Kundenkontakt und die Automobilbranche? Für unseren Kunden, ein international erfolgreiches Unternehmen im Mobilitäts- und Dienstleistungsumfeld, suchen wir einen…

Details Anzeigen
Veröffentlicht am 2026-09-03

Buchhalter (m/w/d) in Direktvermittlung

RADAS Jobbörse & Personalvermittlung GmbH
München

+++ DIREKTVERMITTLUNG in Festanstellung (keine Zeitarbeit) / Vermittlungsgutscheine (AVGS) werden akzeptiert +++ Bei Fragen einfach unverbindlich anrufen: 03048479484 oder einen Rückruf vereinbaren…

Details Anzeigen
Veröffentlicht am 2026-08-04

Senior Softwareentwickler- Autosar (m/w)

AllAtNet Recruiting
München

- Verantwortlich für die Durchführung von Softwareentwicklungsprojekte nach Autosar- Standard - Entwicklung von Steuergeräten - Entwicklung und Integration von Embedded Softwaremodulen in ein Embedd…

Details Anzeigen
Veröffentlicht am 2025-07-06

Praktikant Automobildesign MINI (w/m/x)

München

Unser Team im MINI Designstudio der BMW Group beschäftigt sich mit allen Aspekten der Fahrzeuggestaltung. Wir widmen uns Konzept- und Serienkarosserie, Interieur-Entwürfen sowie der Detailgestaltun…

Details Anzeigen
Veröffentlicht am 2025-12-18

Junior-Bauleiter (w/m/d)

GKK Ingenieurgesellschaft für Hochbau mbH
München

Darauf darfst Du Dich freuen: Neben einer angenehmen, kollegialen Atmosphäre findest Du bei uns ein dynamisches Arbeitsumfeld in einem innovativen Unternehmen. Als Teil unseres hochmotivierten Tea…

Details Anzeigen
Veröffentlicht am 2026-07-21

Showroom-Manager m/w/d in München gesucht

München

Wir suchen in München für unseren Kunden einen Showroom-Manager m/w/d. Die Stelle ist in Vollzeit zu besetzen mit der Option zur Übernahme. ~ Wir bieten Ihnen auch gut ausgestattete Wohnmöglich…

Details Anzeigen
Veröffentlicht am 2025-11-12

TEAM LEAD BARISTA / CAFÉ SUPERVISOR (M/W/D)

care4coffee GmbH
München

Wir sind ALRIGHTY . Und wir lieben richtig guten Kaffee. Nach über 30 Jahren im Kaffeehandwerk sind wir angetreten, um die Kaffeewelt ein bisschen besser zu machen. Tag für Tag. Tasse für Tasse. …

Details Anzeigen
Veröffentlicht am 2026-08-31

Technical Assistant (m/f/d) in Translational Animal Biotechnology

Universitätsklinikum der Technischen Universität München AöR
München

Technical Assistant (m/f/d) in Translational Animal Biotechnology Full- and Part-time | terminated (12 months) | TUM Hospital Animal models of human disease are essential for studying disease p…

Details Anzeigen
Veröffentlicht am 2026-08-26

Visiting VC Analyst - Intern Venture Capital (m/f/d)

Alstin Capital
München

Alstin Capital a Munich-based early-stage venture capital fund focused on B2B software. We invest initial tickets of EUR 2-8M in fast-growing Seed and Series A B2B software companies across Europe…

Details Anzeigen
Veröffentlicht am 2026-08-24