Job Description
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.ResponsibilitiesDesign and optimize high-performance kernels for TPU and GPU architectures using low-level programming frameworks such as Pallas, Triton or MosaicBuild and maintain ...