We're looking for a Lead Kernel Engineer/Architect to join our team in Switzerland 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.

Responsibilities

• Design and optimize high-performance kernels for TPU and GPU architectures using low-level program...