Senior Software Engineer – ML Infrastructure (Python)

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

Were 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 Arms 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 Youll Do

In this role, youll 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 wont be writing individual tests - youll 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 doesnt 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 dont tick every technical box, wed 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 individuals application e.g. visa application fee and Immigration Health Surcharge, will be the responsibility of the successful candidate.

#li-cm1

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