Senior Software Engineer – ML Infrastructure (Python)

In this role, you design and evolve frameworks to exercise, measure, and explain ML workloads at scale across Arm platforms. You’ll build tooling for ML validation and performance analysis, collaborating with ML software, systems, and IP teams to drive improvements. Your work enables large-scale validation on board farms and pre-silicon environments, advancing Arm’s ML performance capabilities. This is a hands-on engineering role focused on scalable, trusted workflows that support timely, high-quality ML releases.

• hybrid working
• accommodations during recruitment
• support for diverse and inclusive environment
• equal opportunity employer
• sponsorship for skilled workers (where applicable)

• Develop frameworks and tooling to run ML workloads and benchmarks across heterogeneous Arm platforms
• Automate execution, data collection, and visualization pipelines for quality and performance assessment
• Collaborate with ML software, systems, and IP teams to define requirements and drive improvements
• Explore ML performance on future Arm architectures and pre-silicon platforms
• Design systems that enable large-scale ML validation and performance analysis

• 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 embrace of diversity
• Ownership mindset and cross-team collaboration
• Understanding of operating system fundamentals (processes, file systems, resource management)
• Experience with pre-silicon or hardware bring-up environments
• Experience running workloads on FPGAs, mobile devices, or development boards