Staff/Senior Software Engineer - Machine Learning Compilers (Some experience required)

Arms Machine Learning (ML) Group is seeking a highly motivated and creative 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.

Job Description:
Arm Machine Learning (ML) Compilers 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 TensorFlow, IREE 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
● 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 teams delivery and lead others on large or more sophisticated tasks
● Good interpersonal and communication skills.

Nice to have skills and experience:
● Technical leadership 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 mentality.

#LI-JB1

Accommodations at Arm:
At Arm, we want our people to Do Great Things. If you need support or an accommodation to Be Your Brilliant Self during the recruitment process, please email [email protected]. To note, by sending us the requested information, you consent to its use by Arm to arrange for appropriate accommodations. All accommodation requests will be treated with confidentiality, and information concerning these requests will only be disclosed as necessary to provide the accommodation. Although this is not an exhaustive list, examples of support include breaks between interviews, having documents read aloud or office accessibility. Please email us about anything we can do to accommodate you during the recruitment process.

Hybrid Working at Arm:
Arm’s approach to hybrid working is designed to create a working environment that supports both high performance and personal wellbeing. We believe in bringing people together face to face to enable us to work at pace, whilst recognizing the value of flexibility. Within that framework, we empower groupsteams to determine their own hybrid working patterns, depending on the work and the team’s needs. Details of what this means for each role will be shared upon application. In some cases, the flexibility we can offer is limited by local legal, regulatory, tax, or other considerations, and where this is the case, we will collaborate with you to find the best solution. Please talk to us to find out more about what this could look like for you.

Equal Opportunities at Arm:
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.