Senior Software Engineer – AI Compiler

Overview

In this role you design and implement an AI/ML compiler within Arm’s Central Technology ML group, aligning with the ML software stack. You will tackle model ingestion, low-level hardware programming, and optimization to improve performance on Arm platforms. You’ll collaborate with specialists across the company and solve complex compiler and tooling challenges. This is a chance to influence AI workloads at scale and shape the future of ML on Arm hardware.

Pay / Benefits

• generous stock awards
• competitive reward package
• benefits supporting wellbeing

Responsibilities

• Design and implement AI/ML compiler components
• Ensure smooth integration with the ML software stack
• Collaborate with cross-functional partners across the company
• Tackle complex problems related to compiler performance and reliability
• Adapt to a fast-evolving environment and contribute to knowledge growth

Key requirements

• Software lifecycle experience from design to production and support
• Strong programming in C++ and Python
• Compiler knowledge, with MLIR or TVM experience
• Rigorous software testing and design for testability
• Ability to work independently or in small teams with initiative
• Fluency in English and good communication skills

• Collaboration
• Initiative
• Adaptability
• C++
• Python
• MLIR or TVM

…