On-device ML Infrastructure Engineer (Core ML Framework)
Apple
On-site
Cupertino, California, United States
Machine Learning Engineer
Our group is seeking an ML Infrastructure Engineer, with a focus on the development of Apple’s Core ML framework. The role entails designing, implementing and maintaining APIs for on-device model execution.
We are building an end-to-end developer experience for ML development that, by taking advantage of Apple’s vertical integration, allows developers to iterate on model authoring, optimization, transformation, execution, debugging, profiling and analysis. This role focuses on the Core ML framework APIs for execution on-device. We’re looking for a highly motivated software engineer that is creative, talented, and passionate about providing high quality developer tools and APIs in the fast paced and dynamic space of ML
- BS/MS/PhD in Computer Science or Electrical Engineering
- Solid programming skills in at least one of the following programming languages: Swift,
- Objective-C, C/C++, Rust with strong attention to detail
- System level programming and debugging experience such as efficient file I/O, memory management, and concurrency.
- Experience building, maintaining, and evolving software libraries
- Experience with API design and software architecture.
- Experience with on-device ML frameworks (Core ML, Win ML, ONNX, TF Lite or ExecuTorch)
- Knowledge of general ML Framework implementation (Jax, PyTorch, or TensorFlow)
- Experience with MLIR / LLVM compiler technologies
- Exceptional verbal and written communication skills