Senior Software Engineer - AI Compiler

Job Title

Senior Software Engineer - AI Compiler

Role Summary

Develop and maintain an AI/ML compiler targeting Arm platforms. Work on model ingestion, low-level hardware programming, memory allocation and performance optimisation, and ensure the compiler integrates with the ML software stack.

Collaborate with specialists across the company to deliver production-quality compiler features for ML workloads.

Experience Level

Senior (title indicates senior level). No explicit years-of-experience requirement listed.

Responsibilities

Primary responsibilities include implementing compiler functionality, ensuring correctness and performance, and supporting production systems.

• Design and implement ML graph compilation components and passes.
• Integrate compiler components with the broader ML software stack and toolchain.
• Implement model ingestion, memory allocation strategies and low-level hardware programming.
• Profile and optimise performance for ML and GenAI workloads.
• Develop and maintain robust tests and frameworks for compiler validation.
• Support production releases and provide ongoing maintenance and troubleshooting.
• Collaborate with hardware, runtime and systems teams to meet cross-functional goals.

Requirements

Essential technical skills and site eligibility are listed below. Candidates must be able to work in the UK without sponsorship.

Must-have:

• Strong programming skills in C++ and Python.
• Experience across the software lifecycle: design, production release and support.
• Compiler knowledge; ideally experience with ML compilation technologies such as MLIR or TVM.
• Ability to design and execute rigorous software tests and testing strategies.
• Capable of working independently and in small teams with initiative.
• Fluency in written and spoken English and effective communication skills.
• Must have the right to work in the United Kingdom without sponsorship.

Nice-to-have:

• Experience with software simulators and/or FPGAs.
• Exposure to driver development, inference engines (e.g. llama.cpp, vLLM), or performance analysis of ML/GenAI workloads.
• Familiarity with NPUs or GPUs and how they are used for ML.

Education Requirements

Not specified.

About the Company

Company: Arm

Headquarters: Cambridge, United Kingdom

ARM is a global leader in semiconductor and software design, driving innovation in computing technology. The company specializes in designing processors and systems that provide the essential building blocks for electronic devices. ARM's architecture is widely used in smartphones, servers, and IoT devices, and its collaborative culture fosters bold thinking, diversity, and high-impact benefits for its talented workforce.