Software Machine Learning Engineer

Job Title

Software Machine Learning Engineer

Role Summary

Develop software for machine learning systems across the ML software stack, including compilers, frameworks, drivers, real-time embedded software, test automation, and performance evaluation. Work in small cross-functional teams and collaborate with ML hardware engineers to optimise system partitioning.

Expectations include delivering technical features, leading larger work packages, and mentoring junior engineers.

Experience Level

Mid-level (experienced engineer). Specific years of experience not stated.

Responsibilities

Primary responsibilities include design, implementation and delivery of ML software components and close collaboration with hardware teams.

• Design and implement features for ML compilation tools and framework software.
• Develop kernel and user-space driver stacks and embedded real-time software where required.
• Create and maintain test automation and performance evaluation suites.
• Lead larger work packages, manage milestones and dependencies across teams.
• Mentor junior engineers and contribute to technical reviews and architecture decisions.
• Collaborate with ML hardware engineers to optimise system partitioning and performance.

Requirements

Must-have skills and relevant experience required to perform the role. Nice-to-have items commonly found on successful candidate profiles are listed separately.

• Must-have: Strong programming skills in C and C++; experience with Python for tooling or automation.
• Proven development experience on cross-team projects with milestones and dependencies.
• Source control experience (Git, Gerrit, Repo).
• Familiarity with agile software development, continuous integration and test-driven development.
• Ability to analyse complex problems and propose practical solutions; good interpersonal and collaboration skills.

• Nice-to-have: Experience with machine learning applications and ML frameworks.
• Experience in compiler development or optimisation.
• Open source software development experience.
• Awareness of hardware architecture principles.

Education Requirements

Not specified.

About the Company

Company: Arm

Headquarters: Cambridge, United Kingdom

ARM is a global leader in semiconductor and software design, driving innovation in computing technology. The company specializes in designing processors and systems that provide the essential building blocks for electronic devices. ARM's architecture is widely used in smartphones, servers, and IoT devices, and its collaborative culture fosters bold thinking, diversity, and high-impact benefits for its talented workforce.