Senior Software Engineer - Machine Learning

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

This role presents an opportunity to contribute to advance ML compilation technology. You will help to build the software that enables development of deep learning applications that form the basis of many ground breaking technologies like self driving cars, generative AI engines and ML-powered wearables.

Arm Machine Learning (ML) team is looking for a software engineer who would build a range of innovative compiler solutions for a variety of markets.

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

If you are interested in this opportunity, make sure to apply soon! We look forward to receiving your application and welcoming you to Arm. You could be joining our highly motivated team and have a marked impact on both strategy and implementation!

Responsibilities

• Contribute to deliver production grade software and push the boundaries of Machine Learning compilation
• Build, extend and collaborate on innovative ML compilation software projects, such as 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. Experience with the full software development lifecycle - planning, designing, developing, testing, delivering, and maintaining production quality software
• Experience with C++, understanding of Python is a plus
• Experience with or interest in compilers such as LLVM and the MLIR ecosystem
• High degree of initiative and problem solving skills
• Ability to own team's delivery and lead others on large or more sophisticated tasks
• Good interpersonal and communication skills

Nice to have skills and experience

• Knowledge or curiosity about large language models (LLMs), machine learning, their applications and frameworks
• Experience with contributing to open source projects and working with a broader open source community
• Experience with Python packaging, 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 mentality.

Experience Level

Senior (5+ years of experience)

Equal Opportunities

Arm is an equal opportunity employer, committed to providing an environment of mutual respect where equal opportunities are available to all applicants and colleagues. We are a diverse organization of dedicated and innovative individuals, and don't discriminate on the basis of race, color, religion, sex, sexual orientation, gender identity, national origin, disability, or status as a protected veteran.