Apple is looking to solve the business and technical problem of enabling the Research to Production lifecycle of cutting-edge machine learning models that power user experiences on Apple's hardware and software platforms by developing critical infrastructure for on-device machine learning.
Requirements
- 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
Responsibilities
- Designing and implementing APIs supporting on-device deployment of ML models
- Architecting and maintaining the internals of the Core ML framework
- Evolving the Core ML model format
- Working with internal and external teams to enable their use cases powered by on-device ML
- Triaging and addressing framework performance issues and functionality gaps
- Building critical infrastructure that begins with onboarding the latest machine learning architectures to Apple devices
- Optimization toolkits to optimize these models to better suit the target devices
Other
- BS/MS/PhD in Computer Science or Electrical Engineering
- Exceptional verbal and written communication skills
