1 month ago
Cambridge, United KingdomSenior
Responsibilities
- Design and implement an AI/ML compiler and integrate it with the broader ML software stack.
- Contribute to ML graph compilation across model ingestion, low-level hardware programming, memory allocation, and performance optimization.
- Take complex software through design, production, and support across the full software lifecycle.
- Collaborate with specialists across the company and work independently or in small groups.
Requirements
- Strong programming expertise in C++ and Python.
- Compiler knowledge, ideally including ML compilation technologies such as MLIR or TVM.
- Experience taking complex software from initial design through production and support.
- Understanding of rigorous software testing and designing software for testability.
- Ability to use AI tools effectively when working on a shared codebase.
- Fluency in written and spoken English, strong communication skills, initiative, adaptability, and a collaborative mindset.
- Preferred: experience with software simulators or FPGAs, driver development, llama.cpp, vLLM, ML and GenAI workload performance analysis, NPUs, or GPUs.
Benefits
- Competitive reward package including generous stock awards and a range of wellbeing benefits.
- Open, accepting, and collaborative environment supporting professional and personal growth.
- Hybrid working, with team-specific working patterns and role details provided during the application process.
Categories
About Arm
Arm’s foundational technology is defining the future of computing. A future built by the greatest technology ecosystem in the world. A future built on Arm. Arm is everywhere technology matters. Technology matters everywhere. Together, we’ll power every technology revolution moving forward, including cloud computing, automotive and autonomous systems, IoT, the metaverse, and beyond. Changing the world. Again. On Arm.