Company: ARM
Location: Cambridge
Job Description:
Overview
In this role you will develop and optimize C++ and Python software for ML workloads on Arm platforms, collaborating with ML software and hardware engineers to improve performance and efficiency. You’ll work across ML frameworks, runtimes, drivers, and embedded software, tackling complex engineering challenges that bridge software and hardware. The position offers growth through technical leadership and opportunities to influence how AI workloads run on future compute platforms.
Pay / Benefits
- comprehensive range of employee benefits
- career development
- collaborative working environment
- hybrid working
- support for accommodations
Responsibilities
- Develop and optimize C++ and Python software for ML workloads
- Collaborate with ML software and hardware engineers to understand model execution and performance bottlenecks
- Contribute to ML compilation, PyTorch integration, runtimes and drivers, and performance analysis
- Engage in embedded software work and performance optimization
- Take ownership of engineering challenges and support development of other engineers
Key requirements
- Professional C++ development experience
- Experience with ML frameworks or tooling such as PyTorch
- Python for software development, ML tooling, testing or automation
- Debugging and performance analysis skills
- Collaborative software development using source control, code review, testing and CI
- collaborative mindset
- problem solving
- ability to learn across ML software stack
- C++
- PyTorch
- Python
…
Posted: September 21st, 2026