On-Device ML Engineering Manager (ML Authoring u0026 Conversion Tools)

Apple • Cupertino, CA

74d

About The Position

This is a fantastic opportunity to join the Machine Learning team at Apple, in a leadership role. The On-Device Machine Learning team at Apple enables the Research to Production lifecycle of innovative machine learning models that power magical user experiences on Apple's hardware and software platforms. Apple is the best place to do on-device machine learning, and this team sits at the heart of that field, collaborating with research, SW engineering, HW engineering, and products. This team builds critical infrastructure, beginning with onboarding the latest machine learning architectures to embedded devices, optimization toolkits to optimize these models to better suit the target devices, machine learning compilers and runtimes to implement these models as efficiently as possible, and the benchmarking, analysis, and debugging toolchain needed to improve on new model iterations. This infrastructure underpins most of Apple's critical machine learning workflows across many of our products, including Camera, Siri, Health, Vision, and is a significant part of Apple Intelligence. We are seeking an Engineering Manager to oversee the team which builds the Authoring and Conversion toolchain. This is one of the key functional areas in our platform, responsible for developing the foundational authoring and conversion APIs that enable the authoring and import of ML models into Apple's ML deployment platform. The team works with foundational and applied machine learning researchers and engineers to onboard the newest architectures. More specifically, the team owns the Python authoring toolchain and APIs, the MLIR front-end compiler and intermediate representations, and the integration with ML research frameworks.

Requirements

• Deep technical understanding of ML research to production workflows.
• Experience with ML compilations and runtimes.
• Knowledge of on-device optimization techniques such as quantization and distillation.
• Familiarity with Apple framework and API design patterns.
• Experience with on-device ML frameworks (Core ML, Win ML, ONNX, TF Lite, or ExecuTorch).
• Experience with MLIR / LLVM compiler technologies.

Responsibilities

• Oversee the team that builds the Authoring and Conversion toolchain.
• Develop foundational authoring and conversion APIs for ML models.
• Collaborate with foundational and applied machine learning researchers and engineers.
• Manage the Python authoring toolchain and APIs.
• Handle the MLIR front-end compiler and intermediate representations.
• Integrate with ML research frameworks.
• Support a dynamic team of engineers with diverse skills.
• Partner with other leaders across the organization.

What This Job Offers

• Career Level: Manager
• Industry: Computer and Electronic Product Manufacturing
• Number of Employees: 5,001-10,000 employees