Machine Learning Software Engineer (C++ & Python)

Job Title

Machine Learning Software Engineer (C++ & Python)

Role Summary

Develop and optimise C++ and Python software components that improve performance and efficiency of machine learning workloads on Arm compute platforms. Work spans ML frameworks, compilers, runtimes, drivers and embedded software.

The role sits within an ML engineering team collaborating with ML software and hardware engineers to identify performance bottlenecks and implement practical solutions across the ML software stack.

Experience Level

Mid-level (hiring across several seniority levels; specific seniority and scope will be matched to candidate experience).

Responsibilities

Deliver software and engineering contributions across the ML software stack. Typical responsibilities include:

• Design, implement and optimise C++ components for ML workloads.
• Develop Python tooling, tests and automation related to ML software and integration tasks.
• Integrate and optimise frameworks such as PyTorch and related runtime components.
• Profile and diagnose performance issues; implement performance improvements.
• Work on ML compilation, runtimes, drivers or embedded software depending on project needs.
• Contribute to technical design, take ownership of larger engineering tasks and support the development of other engineers.
• Collaborate using source control, code review, and continuous integration workflows.

Requirements

Must-have technical skills and experience:

• Professional experience in C++ development on complex software systems.
• Practical experience with Machine Learning frameworks or tooling such as PyTorch.
• Python experience for development, tooling, testing or automation.
• Strong debugging and performance-analysis skills for software systems.
• Experience working in collaborative development environments: source control, code review and CI.

Nice-to-have:

• Knowledge of ML compilers, PyTorch internals, runtimes, drivers or embedded software.
• Experience with hardware acceleration and performance optimisation for ML workloads.

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.