Staff Machine Learning Architect - Assembly Coding and Performance Engineer

ARM
Cambridge
1 week ago
Applications closed

Related Jobs

View all jobs

Staff Machine Learning Engineer

Staff Machine Learning Engineer (NLP, LLMs) - new PurpleAI product

Principal Machine Learning Engineer - Chat UnitedKingdom ...

Senior Solutions Architect

Staff Data Scientist

Principal Software 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 toDo Great Things. If you need support or an accommodation toBe Your Brilliant Selfduring 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.

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.

When Qubits Fuel Neural Networks: The Emerging Frontier of Quantum-Enhanced AI

Artificial Intelligence (AI) has soared to unimaginable heights over the last few years, revolutionising sectors ranging from healthcare and finance to logistics and entertainment. But while deep learning models have grown more sophisticated and powerful, they remain tied to the limitations of classical computing systems. Enter quantum computing, a burgeoning field that leverages qubits—quantum bits—to achieve processing speeds that could leave even today’s most advanced supercomputers in the dust. What if we combined these two forces? Quantum-enhanced AI aims to integrate quantum hardware and algorithms into AI workflows, potentially unlocking efficiencies and capabilities that are currently out of reach. Although this domain is still in its infancy, experts predict it could reshape entire industries in the not-so-distant future. For professionals in AI, this is more than just an interesting development; it’s a pivotal shift that could spawn new roles, research areas, and opportunities. In this thought-leadership piece, we will: Outline the basics of quantum computing and why it matters to AI. Examine how quantum resources might supercharge neural networks. Highlight the career paths at the intersection of quantum and machine learning. Discuss the long-term outlook and what it means for AI professionals looking to stay ahead. Whether you’re already immersed in AI or just beginning to explore its potential, strap in—this new frontier promises a radical transformation.

AI Jobs at Newly Funded UK Start-ups: Q3 2025 Investment Tracker

Over the last decade, the United Kingdom has firmly established itself as one of Europe’s most significant technology hubs. Thanks to a vibrant ecosystem of venture capital, government-backed initiatives, and a wealth of academic talent, the UK has become especially fertile ground for artificial intelligence (AI) innovation. This growth is not just evident in established tech giants—new start-ups are emerging every quarter with fresh ideas, ground-breaking technologies, and a drive to solve real-world problems. In this Q3 2025 Investment Tracker, we take a comprehensive look at the latest AI start-ups in the UK that have successfully secured funding. Beyond celebrating these companies’ milestones, we’ll explore how these recent investments translate into exciting new job opportunities for AI professionals. Whether you’re an experienced machine learning engineer, a data scientist, or simply hoping to break into the AI sector, this roundup will give you insights into the most in-demand roles, the skills you need to stand out, and how you can capitalise on the current AI hiring boom.