Senior AI Compiler Engineer - ML Graph & Optimization

Arm; Cambridge; Hybrid; GBP 74,000 - 100,000; Full time

Job summary

Arm is seeking skilled software engineers to join the ML compiler development team in Cambridge. You will design and implement an AI/ML compiler, ensuring smooth integration with the ML software stack and collaboration with diverse experts across Arm.

The role emphasizes learning, adaptability, and problem solving, with a hybrid working model and a competitive compensation package including stock awards.

Qualifications

• Experience with the full software lifecycle from design to production and support.
• Strong programming expertise in C++ and Python.
• Knowledge of ML compilation technologies such as MLIR or TVM.
• Understanding of rigorous testing and testable software design.
• Ability to work independently or in small groups with initiative.
• Fluency in English, both written and spoken.
• Good communication skills and collaborative mindset.

Responsibilities

• Contribute to the design and implementation of an AI/ML compiler.
• Ensure integration with the ML software stack.
• Collaborate with specialists across Arm to shape AI on Arm platforms.

Skills

C++

Python

ML compilation

Software lifecycle

English communication

Independent work

Collaboration

Tools

MLIR

TVM

FPGA/simulators

Job description

Arm is seeking skilled software engineers to join the ML compiler development team in Cambridge. You will design and implement an AI/ML compiler, ensuring smooth integration with the ML software stack and collaboration with diverse experts across Arm.

The role emphasizes learning, adaptability, and problem solving, with a hybrid working model and a competitive compensation package including stock awards.