Software ML Engineer

Job Title

Software ML Engineer

Role Summary

Develop and deliver software across the ML stack including compilation technology, framework software, kernel and user-space drivers, and real-time embedded components. Work with ML hardware engineers to optimise system partitioning and performance; lead larger work packages and mentor junior engineers within small cross-functional teams.

Experience Level

Mid-level. The listing expects an experienced engineer able to lead larger work packages and mentor junior staff; no explicit years-of-experience requirement provided.

Responsibilities

Key responsibilities include:

• Design and implement features in ML compilation technology and framework software.
• Develop kernel and user-space drivers, real-time embedded software, and supporting infrastructure.
• Create and maintain test automation and performance evaluation tooling.
• Collaborate with ML hardware teams to define optimal system partitioning and performance targets.
• Lead cross-team work packages, manage dependencies and milestones, and mentor junior engineers.

Requirements

Must-have and preferred skills:

• Must-have: Strong programming skills in C, C++ and Python.
• Must-have: Experience working on cross-team projects with milestones and dependencies.
• Must-have: Source control experience (Git, Gerrit, Repo).
• Must-have: Familiarity with agile development practices, continuous integration and test-driven development.
• Nice-to-have: Experience with machine learning applications and frameworks.
• Nice-to-have: Compiler development experience.
• Nice-to-have: Open-source software development experience.
• Nice-to-have: 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.