ML AI Compiler Engineer - C++/Python Expert

CamWebDir; Cambridge; Hybrid; GBP 90,000 - 130,000; Full time

Benefits offered by this job

Stock awards

Wellbeing benefits

Accommodations for recruitment

Job summary

Arm's Central Technology Machine Learning Group in Cambridge is seeking software engineers to design and implement an AI/ML compiler and ensure smooth integration with our ML software stack.

The role requires strong C++ and Python skills, experience across the software lifecycle, and the ability to work independently or in small groups. This is a hybrid role in the UK with opportunities to contribute to innovative ML workloads.

Qualifications

• Experience with the whole software lifecycle from design to production and support.
• Strong programming in C++ and Python.
• Knowledge of ML compilation technologies (MLIR/TVM) and compilers.
• Ability to test software and design for testability.
• Ability to work independently or in small teams with initiative.
• Fluency in English, both written and spoken.

Responsibilities

• Contribute to the design and implementation of an AI/ML compiler.
• Ensure integration with the ML software stack and collaboration across teams.

Skills

C++

Python

ML compilation

MLIR

TVM

SDLC

English fluency

Job description

Arm's Central Technology Machine Learning Group in Cambridge is seeking software engineers to design and implement an AI/ML compiler and ensure smooth integration with our ML software stack.

The role requires strong C++ and Python skills, experience across the software lifecycle, and the ability to work independently or in small groups. This is a hybrid role in the UK with opportunities to contribute to innovative ML workloads.