Staff Software ML Engineer

Job Title

Staff Software ML Engineer

Role Summary

Senior engineering role on Arm's ML software stack focusing on ML compilation, framework software, drivers, embedded runtime, test automation, and performance evaluation. Work in small, cross-functional teams and collaborate closely with ML hardware engineers to optimize system partitioning.

Expectations include leading substantial work packages, mentoring junior engineers, and delivering pragmatic, high-quality solutions for customers building accelerator designs.

Experience Level

Senior — staff-level role (senior technical contributor/lead). Specific years of experience not stated.

Responsibilities

You will design and implement features across the ML software stack, own large work packages, and collaborate across teams to deliver production-quality software.

• Design and develop ML compilation technology and framework software.
• Develop kernel and user-space driver stacks and embedded runtime components.
• Create and maintain test automation and performance evaluation infrastructure.
• Work closely with ML hardware teams to define optimal hardware/software partitioning.
• Lead larger projects, define milestones, and coordinate dependencies across teams.
• Mentor and review work of junior engineers; promote engineering best practices.

Requirements

Critical skills and experience required for the role, followed by desirable skills.

• Proven development experience on multi-team projects with milestones and dependencies.
• Strong programming skills in C, C++, and Python.
• Experience with source control tools such as Git, Gerrit, and Repo.
• Familiarity with agile development, continuous integration, and test-driven development.
• Strong analytical skills and experience working in fast-paced, collaborative environments.

Nice-to-have

• Experience with machine learning applications and ML frameworks.
• Experience in compiler development.
• 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.