Lead Kernel Architect for AI Accelerators

EPAM Systems; Germany (OH); Hybrid; USD 104,919 - 151,550; Full time

Job summary

EPAM Systems is seeking a Lead Kernel Engineer/Architect to join our team in Germany, working in a hybrid model. This role involves designing high-performance kernels for TPU and GPU architectures, while pushing advanced hardware accelerators to their limits.

The ideal candidate will have extensive experience in software engineering and performance optimization, particularly at the kernel level. Interested professionals should apply to be a part of our mission to enhance AI performance and scalability.

Qualifications

• 12+ years of industry experience in software engineering or systems programming.
• 5+ years of experience in software development using C++ or Python.
• Hands-on experience in performance optimization at the kernel level for accelerators.

Responsibilities

• Design and optimize high-performance kernels for TPU and GPU architectures.
• Build and maintain performance infrastructure including benchmarking and regression testing.
• Collaborate with ML framework developers and compiler teams for kernel integration.

Skills

C++

Python

Performance optimization

Kernel programming

Education

Bachelor's degree or equivalent practical experience

Tools

CUDA

Triton

Pallas

Job description

EPAM Systems is seeking a Lead Kernel Engineer/Architect to join our team in Germany, working in a hybrid model. This role involves designing high-performance kernels for TPU and GPU architectures, while pushing advanced hardware accelerators to their limits.

The ideal candidate will have extensive experience in software engineering and performance optimization, particularly at the kernel level. Interested professionals should apply to be a part of our mission to enhance AI performance and scalability.