Edge AI Software Engineer: Compiler & Runtime Optimisation

Arm is seeking experienced software engineers to help develop ExecuTorch, PyTorch’s framework for efficient AI model execution on edge devices. You will work across Python tooling, ahead-of-time compilation, and portable C++ runtimes with hardware-accelerated backends, optimizing AI workloads on Arm Mali GPUs and Cortex CPUs/NPUs.

The role involves moving between high-level model transformations and low-level runtime and hardware behavior, identifying bottlenecks, and delivering practical