Solution Architect - Kernels and Performance, Core ML

We're looking for a Solution Architect – Kernels and Performance , Core ML to join our team in Lisbon, Portugal, in a hybrid working mode. In this role, you will drive low-level performance engineering of AI workloads, optimizing both model training and inference across advanced accelerator architectures like TPU and GPU. You will work on cutting-edge ML models, toolchains and frameworks, enabling scalable, efficient deployment of AI solutions in production. This position combines deep system-level engineering with architectural leadership, directly impacting next-generation AI performance.

Responsibilities

• Design and optimize high-performance kernels using low-level languages like Pallas, Mosaic and Triton for TPU and GPU architectures
• Architect infrastructure such as benchmarking suites, autotuning frameworks and performance analysis tools to support kernel development and testing
• Develop regression testing strategies and compr...