Job Description
Arm's Machine Learning (ML) Group is seeking a highly motivated and creative Software Engineer to join and lead a growing team of brilliant engineers located in Cambridge, UK specialising in ML frameworks and compilers.
This role presents an opportunity to contribute to sophisticated ML technology supporting Arm's ML Hardware. You will help to build the software that enables development of deep learning applications in many ground-breaking fields including self-driving cars, generative AI engines and ML-powered wearables.
Job Description:
The ML Tooling team is looking for a software engineer with line management experience who can build a range of innovative compiler solutions for a variety of markets.
You will apply your experience and insight to craft and optimise compilers for machine learning networks that target Arm’s CPUs, GPUs and NPUs.
Responsibilities:
* Contribute to deliver production-grade software and push the boundaries of Machine Learning compilation
* Recruit, develop, support and retain your engineers working as part of a larger team
* Build, extend and collaborate on innovative ML compilation software projects, such as TensorFlow, PyTorch, TOSA and the broader MLIR ecosystem
* Work with other groups in Arm to expand support for Arm architecture and ecosystem
Required skills and experience:
1. A passion for softwar...