Staff Software Engineer – ML (Tooling)

Arm; Cambridge; Hybrid; GBP 97,000 - 132,000; Full time

Benefits offered by this job

Hybrid work model

Job summary

Arm is seeking an experienced and analytical engineer to design and develop software powering ML validation across our next-generation platforms.

You will design and evolve the solutions that exercise, measure, and understand ML software at scale, across Arm’s board farms and pre-silicon environments.

The role emphasizes ownership, learning, and delivering practical improvements, with hybrid working and collaboration across teams.

Qualifications

• Proven experience in software development or automation, with Python required.
• Experience with modern Python development – OOP, and tools like uv and ruff.
• Familiar with Docker and CI/CD systems (Jenkins, GitLab or GitHub).
• Strong communication, cross-cultural awareness and collaboration.
• Understanding of OS fundamentals – processes, file systems, and resource management.
• Experience with pre-silicon or hardware bring-up environments.
• Experience running workloads on FPGAs, mobile devices, or development boards.
• Understanding ML frameworks, compilers, or model execution flows.
• Background in data visualization or performance analytics.

Responsibilities

• Developing frameworks and tooling for running ML workloads and benchmarks across heterogeneous Arm platforms and board farms.
• Automating execution, data collection, and visualization pipelines to streamline quality and performance assessment.
• Collaborating closely with ML software, systems, and IP teams to define requirements, and drive improvements.
• Exploring and influencing ML performance on future Arm architectures and next-generation pre-silicon platforms.
• Designing systems that enable large-scale ML validation and performance analysis.

Skills

Python

CI/CD

Object-oriented design

os fundamentals

Hardware bring-up

ML frameworks

Data visualization

Performance analytics

Tools

Docker

Jenkins

GitLab

GitHub

Job description

Do you want to shape the frameworks that ensure the world’s ML workloads run successfully on Arm technology?

We’re looking for an experienced and analytical engineer to design and develop software that powers ML validation across our next-generation platforms.

About The Team

We are a team of engineers passionate about understanding and improving Machine Learning performance across Arm’s diverse ecosystem.

Our mission is to develop scalable, efficient, and trusted workflows that enable ML software to release on time and to quality, across CPUs, GPUs, NPUs, and emerging compute architectures.

We create the frameworks, automation and other tooling that power ML performance analysis, system validation, and large-scale verification across our board farms and pre-silicon environments.

What You’ll Do

In this role, you’ll design and evolve the solutions that allows ML software and systems to be exercised, measured, and understood at scale.

Your Responsibilities May Include

• Developing frameworks and tooling for running ML workloads and benchmarks across heterogeneous Arm platforms and board farms.
• Automating execution, data collection, and visualization pipelines to streamline quality and performance assessment.
• Collaborating closely with ML software, systems, and IP teams to define requirements, and drive improvements.
• Exploring and influencing ML performance on future Arm architectures and next-generation pre-silicon platforms.

You won’t be writing individual tests - you’ll be designing the systems that make large-scale ML validation and performance analysis possible.

About You

The successful engineer will be highly flexible, quick to learn and be motivated by the opportunity to understand and improve the systems enabling functionalities and performance measurements of future Machine Learning solutions using Arm technology. You take ownership, investigate when something doesn’t behave as expected, and work across team boundaries to find practical solutions.

• Strong experience in software development or automation (Python required).
• Experience with modern Python development - object-oriented design and tools such as uv and ruff.
• Familiarity with Docker and CI/CD systems (Jenkins, GitLab, or GitHub).
• Good communication skills, inter-cultural awareness and you embrace diversity.
• Understanding of operating system fundamentals - processes, file systems, and resource management.
• Experience with pre-silicon or hardware bring-up environments.
• Experience about running workloads on FPGAs, mobile phones, or development boards
• Understanding of ML frameworks, compilers, or model execution flows.
• Background in data visualization or performance analytics.

Even if you don’t tick every technical box, we’d love to hear from you.

What matters most is your curiosity, your aim to learn, and your ambition to make an impact.

If skilled worker sponsorship is required, Arm will meet costs associated with sponsorship for the employer only. All fees associated with the individual’s application e.g. visa application fee and Immigration Health Surcharge, will be the responsibility of the successful candidate.

Accommodations at Arm

At Arm, we want to build extraordinary teams. If you need an adjustment or an accommodation 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 or adjustment 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.

Salary Range:

£97,300 - £131,700 per year