Staff Software Engineer - Machine Learning

Arm's Machine Learning (ML) Group is seeking a highly motivated and creative Software Engineer to join and lead a growing team of brilliant engineers located in Cambridge, UK specialising in ML frameworks and compilers.

This role presents an opportunity to contribute to sophisticated ML technology supporting Arm's ML Hardware. You will help to build the software that enables development of deep learning applications in many ground-breaking fields including self-driving cars, generative AI engines and ML-powered wearables.

Job Description:

The ML Tooling team is looking for a software engineer with line management experience who can build a range of innovative compiler solutions for a variety of markets.

You will apply your experience and insight to craft and optimise compilers for machine learning networks that target Arm’s CPUs, GPUs and NPUs.

Responsibilities:

• Contribute to deliver production-grade software and push the boundaries of Machine Learning compilation
• Recruit, develop, support and retain your engineers working as part of a larger team
• Build, extend and collaborate on innovative ML compilation software projects, such as TensorFlow, PyTorch, TOSA and the broader MLIR ecosystem
• Work with other groups in Arm to expand support for Arm architecture and ecosystem

Required skills and experience:

• A passion for software development and quality
• Line management of a small team (4-8 engineers)
• Proven experience with C++, understanding of Python is a plus
• Experience with the full software development lifecycle - planning, designing, developing, testing, delivering, and maintaining production-quality software
• Experience with or interest in compilers
• Desire to learn new skills and technologies and work in a highly motivated team
• High degree of initiative and problem solving skills
• Ability to own team's delivery and lead others on large or more sophisticated tasks
• Excellent communication skills.

'Nice to have' skills and experience:

• Experience with contributing to open-source projects and working with a broader open-source community
• Knowledge or curiosity about computer vision, machine learning, their applications and frameworks
• Experience with Linux and scripting languages, such as shell-scripting

In Return:

On top of the already compelling Life at Arm, we offer strong team culture, learning opportunities, regular career conversations, emphasis on diversity, equity and inclusion and a continuous improvement approach.