AI Software Engineer - C++/Python
Core
Develop and optimize ExecuTorch, PyTorch's framework for running AI models efficiently on edge devices, focusing on compilers, runtimes, and hardware acceleration. (via careerplan.io/jobs/100031426432-ai-software-engineer-cpython-at-arm)
Role type
Senior IC software engineer (AI compilers and edge runtimes)
Builds
Optimized backends for Mali GPUs, Cortex-A/M CPUs, and Ethos-U NPUs; Python APIs and model transformations for resource-constrained platforms.
Domain
AI software, compilers, embedded systems, edge computing
Deliverable
production ML models
Required skills
C++, Python, compilers, runtimes, performance optimization, profiling, embedded systems, hardware acceleration
Preferred skills
PyTorch, ExecuTorch, ahead-of-time compilers, graph transformation, Vulkan, open-source development, technical leadership
Technologies
PyTorch, ExecuTorch, Mali GPUs, Cortex-A, Cortex-M, Ethos-U NPUs, C++, Python, Vulkan
Responsibilities
Develop and optimize components of the ExecuTorch compiler, runtime, and developer tooling; Analyse and profile AI workloads to identify bottlenecks; Develop optimized backends for edge platforms; Identify technical opportunities and take solutions from idea to delivery; Share technical knowledge and mentor other engineers.
