Staff Software Engineer – ML (Tooling)

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.

Essential skills:

• 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).
• Comfortable developing on Linux or Mac.
• Good communication skills, inter-cultural awareness and you embrace diversity.

Desirable:

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

Our 10x Mindset

At Arm, we believe progress happens when people are empowered to think bigger and push beyond what seems possible. Our 10x mindset is about curiosity, ambition and creating impact that will be used by millions. We learn fast, prioritise collaboration and turn bold ideas into real technology. We look for people who are inspired by this way of working and want to grow in an environment where bold ideas are welcomed. Discover how the 10x mindset shapes life at Arm: https://careers.arm.com/10x-mindset

Use of AI in Your Application

At Arm, we embrace the benefits AI can bring, including in recruitment. You may use AI to support your preparation, such as improving the clarity of written responses, practicing interview questions, or learning more about our industry. However, your application and interview responses must reflect your own thinking, experience, and abilities. We want to understand your authentic perspective, creativity, and approach to problem solving. As such, you may not use AI to complete assigned tasks as part of any assessments. If you plan to use AI-supported or assistive technologies as part of a reasonable adjustment, please contact [email protected] so we can support you.

Recruitment Adjustments and Accommodations

At Arm, we want to build extraordinary teams. If you need an adjustment or an accommodation during the recruitment process, please email [email protected]. 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

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

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.