On-device ML Infrastructure Engineer – Core ML Framework
Company | Apple |
---|---|
Location | Cupertino, CA, USA |
Salary | $143100 – $264200 |
Type | Full-Time |
Degrees | Bachelor’s, Master’s, PhD |
Experience Level | Mid Level |
Requirements
- 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.
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 developers to enable their uses cases powered by on-device ML
- Triaging and addressing framework performance issues and functionality gaps
Preferred Qualifications
- 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