National AI Awards 2025Discover AI's trailblazers! Join us to celebrate innovation and nominate industry leaders.

Nominate & Attend

Senior/Staff Software Engineer - Machine Learning Frameworks

ARM
Ely
7 months ago
Applications closed

Related Jobs

View all jobs

Senior Data Engineer (Maximo)

Staff Data Scientist

Principal / Senior Data Scientist

Senior AI | Machine Learning Engineer

Senior AI | Machine Learning Engineer

Data Engineer - Data Infrastructure

Job Overview:

Are you a skilled and motivated engineer with a strong background in optimised system design such distributed systems and high-performance concurrency?

 

We are building the future of computing, on Arm. We want to make Arm-based hardware the natural choice for ML in the datacenter. To that end, we truly believe that major machine learning frameworks used to run AI must be highly performant.

Our team is a diverse, dedicated and inclusive group from all over the world based in Arm's stylish offices at the heart of Manchester. We work on all aspects of researching, developing and delivering highly optimised ML frameworks into the Arm ecosystem across many ML models.

 

This role will directly contribute to key open source ML frameworks such as TensorFlow and PyTorch. In addition, Arm is owner and advocate of the underlying technologies, such as Compute Library, that act as basic building blocks to form the high-quality and performant software. In collaboration with colleagues from Manchester and Cambridge you will work on delivering optimised software for server-class hardware, and integrate it with ML software frameworks and libraries for deployment on our partner's hardware.

 

We work with exciting technology, help to implement new algorithms, and optimise for the latest Arm server hardware. Our work has high impact in the ML ecosystem, with possibility to engage with partners and the community.

Responsibilities:

Your role as a Senior Software Engineer will require you to:

  • Optimise ML software to utilise the full potential of Arm's line of Neoverse cores for datacentre and cloud uses-cases at high core counts.
  • Work with development teams based from compilers and libraries to extend the capabilities of Compute Library to meet the unique demands of ML workloads on servers.
  • Improve Compute Library API to support flawless integration with ML frameworks.
  • Extend ML frameworks to simplify integration with Compute Library

Required Skills and Experience :

  • Software development, with very good programming skills, preferably C++ and Python.
  • Excellent understanding of parallel programming primitives and constructs.
  • Ability to quickly investigate and debug large software frameworks.
  • Optimising code for performance.
  • A real passion for software development.
  • Very good interpersonal, collaboration and communication skills.
  • Curiosity to make a positive impact, both in our team, and in the wider Arm ecosystem.
  • A degree, or higher, in a computational or numerate subject, or experience in a related field.

Note: Training may be provided if you have only a subset of these key skills!

“Nice To Have” Skills and Experience:

The following is a selection of skills used across our projects.

You do not need to have experience with any of these to apply or succeed in your application.

  • Experience with one or more of: NEON, SVE, SVE2, and Arm assembler.
  • Experience with TensorFlow, PyTorch, ONNXRuntime, and/or oneDNN.
  • Open source community citizenship including code commits and reviews.
  • Basic Linux administration, particularly installation and maintenance.

In Return:

You will be joining an outstanding company! We strive to provide an open and accepting environment where you are encouraged to share your ideas and opinions; which enables collective innovation and creativity, and supports your professional and personal growth. In addition, we enjoy 25 days of annual leave as well as progressive parental leave, support for flexible and hybrid working, and many other benefits that Arm offers.

 

#LI-JB1

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.

National AI Awards 2025

Subscribe to Future Tech Insights for the latest jobs & insights, direct to your inbox.

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

Industry Insights

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

Return-to-Work Pathways: Relaunch Your Machine Learning Career with Returnships, Flexible & Hybrid Roles

Returning to work after an extended break can feel like starting from scratch—especially in a specialist field like machine learning. Whether you paused your career for parenting, caring responsibilities or another life chapter, the UK’s machine learning sector now offers a variety of return-to-work pathways. From structured returnships to flexible and hybrid roles, these programmes recognise the transferable skills and resilience you’ve developed, pairing you with mentorship, upskilling and supportive networks to ease your transition back. In this guide, you’ll discover how to: Understand the current demand for machine learning talent in the UK Leverage your organisational, communication and analytical skills in ML contexts Overcome common re-entry challenges with practical solutions Refresh your technical knowledge through targeted learning Access returnship and re-entry programmes tailored to machine learning Find roles that fit around family commitments—whether flexible, hybrid or full-time Balance your career relaunch with caring responsibilities Master applications, interviews and networking specific to ML Learn from inspiring returner success stories Get answers to common questions in our FAQ section Whether you aim to return as an ML engineer, research scientist, MLOps specialist or data scientist with an ML focus, this article will map out the steps and resources you need to reignite your machine learning career.

LinkedIn Profile Checklist for Machine Learning Jobs: 10 Tweaks to Drive Recruiter Interest

The machine learning landscape is rapidly evolving, with demand soaring for experts in modelling, algorithm tuning and data-driven insights. Recruiters hunt for candidates proficient in Python, TensorFlow, PyTorch and MLOps processes. A generic profile simply won’t cut it. Our step-by-step LinkedIn for machine learning jobs checklist covers 10 targeted tweaks to ensure your profile ranks in searches and communicates your technical impact. Whether launching your ML career or seeking leadership roles, these optimisations will sharpen your professional narrative and boost recruiter engagement.

Part-Time Study Routes That Lead to Machine Learning Jobs: Evening Courses, Bootcamps & Online Masters

Machine learning—a subset of artificial intelligence—enables computers to learn from data and improve over time without explicit programming. From predictive maintenance in manufacturing to recommendation engines in e-commerce and diagnostic tools in healthcare, machine learning (ML) underpins many of today’s most innovative applications. In the UK, demand for ML professionals—engineers, data scientists, research scientists and ML operations specialists—is growing rapidly, with roles projected to increase by over 50% in the next five years. However, many aspiring ML practitioners cannot step away from work or personal commitments for full-time study. Thankfully, a rich ecosystem of part-time learning pathways—Evening Courses, Intensive Bootcamps and Flexible Online Master’s Programmes—empowers you to learn machine learning while working. This comprehensive guide examines each route: foundational CPD units, immersive bootcamps, accredited online MSc programmes, funding options, planning strategies and a real-world case study. Whether you’re a software developer branching into ML, a statistician aiming to upskill, or a professional exploring AI-driven innovation, you’ll discover how to build in-demand ML expertise on your own schedule.