Staff Machine Learning Architect - Assembly Coding and Performance Engineer

Cambridge
2 weeks ago
Applications closed

Related Jobs

View all jobs

Staff Machine Learning Engineer

Staff Machine Learning Engineer

Senior Solutions Architect

Principal Software Engineer

Senior Principal Data Scientist, NLP

Senior/Staff Machine Learning Engineer

Job 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 . 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

Get the latest insights and jobs direct. Sign up for our newsletter.

By subscribing you agree to our privacy policy and terms of service.

Industry Insights

Discover insightful articles, industry insights, expert tips, and curated resources.

How to Advertise AI Jobs and List AI Vacancies: Advanced Recruitment Strategies for 2025

In a landscape where artificial intelligence (AI) is rapidly transforming industries—from healthcare and finance to manufacturing and creative fields—employers are in stiff competition to secure the best AI talent. Whether you’re a start-up looking for your first machine learning engineer or a global enterprise planning an AI research lab, knowing how to advertise AI jobs effectively has never been more critical. Below, you’ll find in-depth strategies for crafting compelling AI job adverts, optimising your recruitment funnel, and showcasing your organisation as an employer of choice for top AI specialists. We’ll also explore the importance of salary transparency, the best channels for promoting your AI vacancies, and advanced techniques for nurturing a culture of innovation.

AI Training Jobs: Your Comprehensive Guide to Launching a High-Potential Career

Artificial Intelligence (AI) has evolved from a futuristic concept to a core component of modern business strategy. As organisations increasingly embrace AI-driven systems to stay competitive, the demand for qualified professionals who can develop, implement, and train AI models has skyrocketed. In the UK—and indeed worldwide—there is a pressing need for skilled experts who understand the nuances of AI, from algorithm design to ethical considerations. For anyone seeking to enter this exciting field or pivot into a role focusing on AI training, the opportunities are abundant. This in-depth blog post will explore everything you need to know about AI training jobs, the essential skills you’ll need, the current employment landscape in the UK, and how to future-proof your career in AI.

Rural-Remote AI Jobs: A Breath of Fresh Air in the UK Tech Scene

A New Horizon for AI Professionals For years, conversations around tech careers in the UK have hinged on a central theme: to succeed in artificial intelligence (AI), you must be in or around London (or other big metropolitan areas like Manchester, Bristol, or Edinburgh). But times are changing. Technological leaps and the rise of flexible working are paving the way for AI professionals to live and work well beyond the capital. From the rugged coastlines of Cornwall and Pembrokeshire to the rolling hills of the Yorkshire Dales, we’re witnessing an exciting trend of AI remote countryside roles that allow you to work at the forefront of tech innovation—all while enjoying the tranquillity of rural or seaside living. At ArtificialIntelligenceJobs.co.uk, we’re seeing a marked increase in job postings and applications for these sorts of positions. A growing segment of job seekers is actively searching for “tech jobs by the sea” or “AI remote countryside,” driven by a desire for better work-life balance, lower living costs, and a healthier lifestyle. And it’s not just employees who stand to benefit; employers eager to attract top-tier AI talent are discovering that offering remote or flexible roles widens their candidate pool and enhances diversity. If you’re enticed by the idea of logging off from a day of coding neural networks and taking a stroll along a coastal path—or stepping outside your converted barn in Northumberland to soak in some fresh country air—this article is for you. Below, we’ll explore the benefits and challenges of rural-remote AI jobs, the specific roles best suited for remote work, and how to position yourself for success in this rapidly evolving sector.