Staff Software ML Engineer
ArmJob Title
Staff Software ML Engineer
Role Summary
Develop and optimize C++ and Python software to improve performance and efficiency of machine learning workloads on Arm compute platforms. Work across framework integration, compilers, runtimes, drivers, embedded software and performance optimisation while collaborating with ML software and hardware engineers.
Experience Level
Senior (Staff-level). Specific years of experience are not stated; the role expects substantial engineering experience and technical leadership at the staff level.
Responsibilities
Primary responsibilities include engineering, optimisation and collaboration across the ML software stack.
- Design, implement and optimise C++ and Python components for ML workloads.
- Integrate and contribute to ML frameworks (for example PyTorch) and related tooling.
- Work on ML compilation, runtimes, drivers, embedded software and performance-critical code paths.
- Perform debugging, profiling and performance analysis to locate and resolve software and system bottlenecks.
- Collaborate with hardware and software teams to improve execution efficiency on Arm platforms.
- Contribute to technical design, take ownership of larger engineering challenges and support the development of other engineers.
Requirements
Key skills and experience required or preferred for successful candidates.
Must-have:
- Professional C++ development experience on complex software systems.
- Practical experience with machine learning frameworks or tooling such as PyTorch.
- Proficiency in Python for development, tooling, testing or automation.
- Strong debugging and performance-analysis skills for complex software problems.
- Experience with collaborative development practices: source control, code review, automated testing and CI.
Nice-to-have:
- Experience with ML compilers, PyTorch internals, runtimes or drivers.
- Background in embedded software, performance optimisation or hardware acceleration.
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.
