Staff Software Engineer – AI Compiler

Overview

In this role you will help design and implement an AI/ML compiler for Arm platforms, integrating it with Arm’s ML software stack. You’ll work within a diverse ML compiler team to address model ingestion, low-level hardware programming, and performance optimization at scale. You will collaborate with specialists across the company to solve complex problems and drive impactful AI tooling. This is a growth-focused, cross-functional environment where your work shapes ML workloads on real hardware. You will be part of a mission-driven group building the foundations for efficient AI on Arm.

Pay / Benefits

• generous stock awards
• wellbeing and benefits package
• hybrid working options
• inclusive and growth-focused culture

Responsibilities

• Design and implement an AI/ML compiler and ensure integration with the ML software stack
• Collaborate with cross-functional teams across the company
• Tackle complex problems in a fast-evolving environment
• Contribute to production-ready software and provide support across the lifecycle

Key requirements

• Experience with the full software lifecycle from design to production and support
• Strong programming skills in C++ and Python
• Knowledge of compilers, ideally ML compilation technologies such as MLIR or TVM
• Understanding of rigorous software testing and test-driven design
• Ability to use AI tools when working on shared codebases
• Ability to work independently or in small teams with initiative
• Fluency in English (written and spoken)
• Good communication and collaborative mindset

• Collaborative mindset
• Effective communication
• Initiative and self-motivation
• C++ and Python programming
• Compiler knowledge (MLIR/TVM)
• Software testing and design for testability

…