Lead Kernel Architect for AI Performance & Accelerators

EPAM Systems is looking for a Lead Kernel Engineer/Architect to join our team in Switzerland in a hybrid working mode. This role involves driving the optimization of machine learning operations for large-scale training and inference using cutting-edge hardware like TPUs and GPUs.

The ideal candidate will have extensive experience in both software engineering and performance optimization. Responsibilities include designing high-performance kernels, collaborating with developers, and maintaining performance infrastructure to enable rapid AI deployments.