Senior AI Compiler Engineer for ML Graphs

Arm Limited; Cambridge; Hybrid; GBP 70,000 - 110,000; Full time

Benefits offered by this job

Stock awards

Wellbeing benefits

Job summary

Arm Limited's Central Technology Machine Learning Group is seeking software engineers to join the ML compiler development team in Cambridge. You will contribute to the design and implementation of an AI/ML compiler, ensuring it integrates with the ML software stack and collaborates with specialists across the company.

You will work in a hybrid model, tackling diverse challenges in ML graph compilation, model ingestion, memory allocation, and performance optimisation while enjoying a competitive

Qualifications

• Experience with the full software lifecycle from design through production and support.
• Strong programming in C++ and Python.
• Knowledge of ML compiler tech (MLIR/TVM).
• Ability to test software rigorously and design for test.
• Ability to use AI tools on a shared codebase.
• Capable of working independently or in small groups with initiative.
• Fluent English (written and spoken).
• Good communication and collaboration.

Responsibilities

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

Skills

Full software lifecycle

C++

Python

ML compiler knowledge

MLIR/TVM experience

Testing for software

AI tools usage

Independent working

English fluency

Communication

Tools

None

Job description

Arm Limited's Central Technology Machine Learning Group is seeking software engineers to join the ML compiler development team in Cambridge. You will contribute to the design and implementation of an AI/ML compiler, ensuring it integrates with the ML software stack and collaborates with specialists across the company.

You will work in a hybrid model, tackling diverse challenges in ML graph compilation, model ingestion, memory allocation, and performance optimisation while enjoying a competitive