On-device ML Infrastructure Engineer (Fine-tuning and Customization)
Cupertino, California, United States | Software and Services
Description
We are building an end-to-end developer experience for ML development that leverages Apple’s vertical integration. This allows developers to iterate on model authoring, optimization, transformation, execution, debugging, profiling, and analysis—crucial elements of delivering a seamless user experience. This role focuses on the Create ML framework for machine learning model customization and Apple Intelligence. We seek a creative, talented, and passionate software engineer excited about providing high-quality developer tools, workflows, and APIs in the dynamic ML space.
Key Responsibilities
- Designing and implementing workflows for fine-tuning and adapting ML models.
- Architecting and maintaining the internals of the Create ML framework.
- Collaborating with internal and external developers to enable their use cases powered by on-device ML.
- Triaging and addressing framework performance issues and functionality gaps.
Minimum Qualifications
- Strong software development skills; proficiency in Swift is a plus.
- Experience with machine learning projects, including model training, evaluation, inference, and on-device optimization.
- Passion for software architecture, data structures, and processes.
- Bachelor's degree in Computer Science or Electrical Engineering.
Preferred Qualifications
- Experience with on-device ML frameworks (e.g., Core ML, Win ML, ONNX, TF Lite, ExecuTorch).
- Knowledge of ML frameworks like Jax, PyTorch, or TensorFlow.
- Experience using Apple’s ML frameworks and libraries.
- MS/PhD in Computer Science or Electrical Engineering.
- Excellent communication skills to help others understand code, problems, and ideas.
This job posting is active, and the salary range is between $147,400 and $272,100, depending on skills, qualifications, experience, and location. Apple offers comprehensive benefits, including medical, dental, retirement plans, stock programs, educational reimbursement, and more. Apple is committed to inclusion and diversity, promoting equal opportunity for all applicants.
#J-18808-Ljbffr