Senior Software Engineer - ML Infrastructure (Python)

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

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.