Staff Machine Learning Architect & Assembly Coding and Performance Engineer

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

Staff Machine Learning Architect - Assembly Coding and Performance EngineerARMCambridgeJob Overview: High-performance ML workloads on Arm CPUs requires the co-development of algorithms and highly optimized CPU kernels. In CT-ML (Central Technology, Machine Learning), rapid kernel prototyping is crucial for exploring algorithms and assessing trade-offs between model accuracy and performance. Successful prototypes are essential to drive future CPU architecture development and also deliverables to Central Engineering for final production. Responsibilities: This position is part of a dedicated team within the CT-ML group to focus on analyzing ML workload, rapid prototyping of highly optimized CPU kernels to drive model performance and accuracies. Required Skills and Experience : Strong interest and passion for implementing high-performance kernel code in a dynamic environment. 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 Deep understanding on CPU architecture “Nice To Have” Skills and Experience : Knowledge of ML models and algorithm is a plus Advanced degree or equivalent experience in Computer Architecture and Software are a plus In Return: Arm is committed to global talent acquisition, offering an attractive relocation package. With offices around the world, Arm is a diverse organization of dedicated, creative and highly talented engineers. By enabling a dynamic, inclusive, meritocratic, and open workplace, where all our people can grow and succeed, we encourage our people to share their unrivaled contributions to Arm's success in the global marketplace. LI-TE! Accommodations at Arm At Arm, we want our people to Do Great Things . If you need support or an accommodation to Be Your Brilliant Self during the recruitment process, please email [e-mail] . To note, by sending us the requested information, you consent to its use by Arm to arrange for appropriate accommodations. All accommodation requests will be treated with confidentiality, and information concerning these requests will only be disclosed as necessary to provide the accommodation. Although this is not an exhaustive list, examples of support include breaks between interviews, having documents read aloud or office accessibility. Please email us about anything we can do to accommodate you during the recruitment process. Hybrid Working at Arm Arm’s approach to hybrid working is designed to create a working environment that supports both high performance and personal wellbeing. We believe in bringing people together face to face to enable us to work at pace, whilst recognizing the value of flexibility. Within that framework, we empower groups/teams to determine their own hybrid working patterns, depending on the work and the team’s needs. Details of what this means for each role will be shared upon application. In some cases, the flexibility we can offer is limited by local legal, regulatory, tax, or other considerations, and where this is the case, we will collaborate with you to find the best solution. Please talk to us to find out more about what this could look like for you. Equal Opportunities at Arm Arm is an equal opportunity employer, committed to providing an environment of mutual respect where equal opportunities are available to all applicants and colleagues. We are a diverse organization of dedicated and innovative individuals, and don’t discriminate on the basis of race, color, religion, sex, sexual orientation, gender identity, national origin, disability, or status as a protected veteran.