Apple is seeking an Engineering Manager to lead the team responsible for building the Authoring and Conversion toolchain, a critical component of their on-device machine learning platform. This toolchain enables the onboarding and deployment of innovative machine learning models that power user experiences on Apple's devices, supporting various products and contributing to Apple Intelligence.
Requirements
- Knowledge of ML development and workflows, including at least one authoring framework experience (e.g., PyTorch).
- Solid programming skills in Python, C++ or Swift.
- Experience with MLIR / LLVM compiler technologies
- Experience with on-device ML frameworks (Core ML, Win ML, ONNX, TF Lite or ExecuTorch)
- Deep technical understanding of ML research production workflows, ML compilations, runtimes, and on-device optimization techniques (such as quantization, distillation, etc.).
- Track record of creating clean software architectures, intuitive designs, and high-performance extensible software.
Responsibilities
- Lead the team building the authoring/conversion technologies to quickly onboard new ML models to our ML platform, including contributions to internal and external ML authoring frameworks, like MLX and PyTorch.
- Understand different ML operations, architectures, and graph representations in different authoring frameworks and stay on top of the latest innovations in this space.
- Partnering with multi-functional partners to push the state-of-the-art of on-device ML functionality and performance.
- Planning and implementing multi-year roadmaps for both internal architectures and public facing functionality.
- Engaging with and supporting 1st and 3rd party developers ML powered use cases.
- Develop foundational authoring and conversion APIs that enable the authoring and import of ML models into Apple’s ML deployment platform.
- Own the Python authoring toolchain and APIs, the MLIR front-end compiler and intermediate representations, and the integration with ML research frameworks.
Other
- Two or more years of strong and validated management experience.
- Excellent social skills.
- Attracting, hiring, and guiding the career of ML and software engineers.
- BS/MS/PhD in Computer Science or Electrical Engineering.
- Familiarity with Apple framework and API design patterns
