Posted in

On-device ML Infrastructure Engineer – Core ML Framework

On-device ML Infrastructure Engineer – Core ML Framework

CompanyApple
LocationCupertino, CA, USA
Salary$143100 – $264200
TypeFull-Time
DegreesBachelor’s, Master’s, PhD
Experience LevelMid 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