On-device ML Infrastructure Engineer (ML Modeling Semantics & Representation)
Apple
This job is no longer accepting applications
See open jobs at Apple.See open jobs similar to "On-device ML Infrastructure Engineer (ML Modeling Semantics & Representation)" AnitaB.org.Software Engineering, Other Engineering, Data Science
Cupertino, CA, USA
USD 207,800-312,200 / year + Equity
Posted on Mar 27, 2025
The On-Device Machine Learning team at Apple is responsible for enabling the Research to Production lifecycle of cutting edge machine learning models that power magical user experiences on Apple’s hardware and software platforms. This team sits at the heart of that discipline, interfacing with research, SW engineering, HW engineering, and products. Our group is looking for an ML Infrastructure Engineer, with a focus on ML model semantics, representation, and optimizations. The role is responsible for working with ML research and Applied research engineers to onboard the newest ML architectures to CoreML’s ML model representation, including evolving the representation to support the latest and greatest features in the authored ML program (e.g., PyTorch), and enable the exposure of Apple’s on-device execution capabilities. The role is responsible for building critical “bridging” infrastructure between the most used ML frameworks (e.g., PyTorch) and Apple’s CoreML stack. Key responsibilities: - Develop technologies to quickly onboard new ML models to our on-device stack, including contributions to ML authoring frameworks. - Understand different ML operations, architectures, and graph representations in different authoring frameworks. Keep abreast of latest innovations in this space. - Architect and build CoreML’s model representation that can efficiently represent program semantics from the authored frameworks, while allowing for peak execution performance. - Define and build the user-facing model translation and ingestion abstractions, APIs, and surrounding toolkit to allow seamless model import into Apple’s ML stack. - Perform optimizations such as quantization, operator transformations, etc. to make models more amenable to efficient on-device deployment
This job is no longer accepting applications
See open jobs at Apple.See open jobs similar to "On-device ML Infrastructure Engineer (ML Modeling Semantics & Representation)" AnitaB.org.