Arm

Senior Software Engineer – ML Infrastructure (Python)

Arm
Apply
2 days ago
Cambridge, United KingdomSenior

Responsibilities

  • Develop frameworks and tooling for running ML workloads and benchmarks across heterogeneous Arm platforms and board farms.
  • 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.
  • Investigate ML performance on future Arm architectures and next-generation pre-silicon platforms.

Requirements

  • Strong experience in software development or automation, with Python required.
  • Experience with modern Python development, object-oriented design, uv, and ruff.
  • Familiarity with Docker and CI/CD systems such as Jenkins, GitLab, or GitHub.
  • Comfort developing on Linux or Mac.
  • Good communication skills, intercultural awareness, and an inclusive approach.
  • Desirable experience includes operating system fundamentals, pre-silicon or hardware bring-up environments, workloads on FPGAs, mobile phones or development boards, ML frameworks, compilers, model execution flows, data visualization, or performance analytics.

Benefits

  • Hybrid working patterns are determined by the team, with role-specific details shared during the application process.
  • Arm provides recruitment accommodations and adjustments upon request.
  • If skilled worker sponsorship is required, Arm covers employer sponsorship costs, while the successful candidate is responsible for individual application fees and the Immigration Health Surcharge.

Tech Stack

DockerJenkinsLinuxmacOSPython

Categories

Arm

About Arm

5,001-10,000 employees

Arm’s foundational technology is defining the future of computing. A future built by the greatest technology ecosystem in the world. A future built on Arm. Arm is everywhere technology matters. Technology matters everywhere. Together, we’ll power every technology revolution moving forward, including cloud computing, automotive and autonomous systems, IoT, the metaverse, and beyond. Changing the world. Again. On Arm.

Contact me