14 days ago
Cambridge, United KingdomStaff+
Responsibilities
- Define, design, and deliver production-grade model analysis and optimization tooling for edge developers through web services and desktop applications.
- Provide technical leadership in ML tooling and shape the architecture and roadmap for ML developer tools.
- Work across ML frameworks, runtimes, existing and future hardware, and heterogeneous execution environments.
- Build collaborative relationships with distributed engineering teams, product managers, and UX specialists.
- Drive maintainable software, automated testing, continuous integration, effective engineering processes, and customer feedback.
- Help build and lead a new team working on ML developer tooling and greenfield software products.
Requirements
- Strong understanding of neural-network architecture and model execution, including computational graphs, transformations, operator dispatch, memory planning, backend partitioning, and heterogeneous hardware execution.
- Proven experience with hardware-aware model optimization, including quantization, graph optimization, operator fusion, and precision reduction.
- Experience with multiple ML frameworks, model formats, and inference runtimes such as PyTorch, ONNX, ONNX Runtime, ExecuTorch, TensorFlow, LiteRT, and OpenVINO.
- Experience analyzing, profiling, and debugging ML workloads, including compatibility, performance, latency, memory, model size, and accuracy trade-offs.
- Passion for building high-quality developer tools and intuitive workflows.
- Preferred experience with ML inspection, visualization, profiling, or benchmarking tools; deployment across CPUs, GPUs, and ML accelerators; Linux, containers, Python, or TypeScript.
- Preferred experience building developer-facing CLIs, SDKs, APIs, or web tools and engaging directly with developers to improve products through prototypes, feedback, and analytics.
Benefits
- Opportunity to build and lead a new team working with innovative technologies in an agile environment.
- Opportunity to work on greenfield software products that ship with new Arm hardware on day one.
- Hybrid working patterns are determined by the group or team, with role-specific details shared during the application process.
- Recruitment accommodations are available, including interview breaks, documents read aloud, and office accessibility support.
Tech Stack
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.