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.