Staff ML Kernel Architect – High-Performance CPU (Hybrid)

Women Thrive Magazine; Cambridge; Hybrid; GBP 125,000 - GBP 150,000; Full time

Job summary

A leading technology firm in Cambridge is seeking a Machine Learning Architect to focus on analyzing ML workloads and prototyping optimized CPU kernels. The ideal candidate will have over 4 years of experience in high-performance kernel coding, measuring performance, and creating development frameworks. This role supports a collaborative and dynamic environment, offering an attractive relocation package and a commitment to diversity and inclusion. Opportunities for flexible hybrid working patterns are also available.

Benefits

Attractive relocation package

Flexible hybrid working environment

Diverse and inclusive workplace

Qualifications

• 4 years experience in implementing high-performance CPU kernel with vector and matrix extensions.
• Experience measuring and understanding performance.
• Experience in creating an efficient kernel code development framework including tools and testing.

Responsibilities

• Analyze ML workload to improve model performance.
• Prototype highly optimized CPU kernels.
• Deliver relevant prototypes for future CPU architecture development.

Skills

High-performance kernel code development

Performance measurement

Kernel code framework creation

Deep understanding of CPU architecture

Education

Advanced degree or equivalent experience in Computer Architecture

Job description

A leading technology firm in Cambridge is seeking a Machine Learning Architect to focus on analyzing ML workloads and prototyping optimized CPU kernels. The ideal candidate will have over 4 years of experience in high-performance kernel coding, measuring performance, and creating development frameworks. This role supports a collaborative and dynamic environment, offering an attractive relocation package and a commitment to diversity and inclusion. Opportunities for flexible hybrid working patterns are also available.