Large Machine Learning Model Optimization Engineer

$166,600 - $296,300/Yr

Apple - Seattle, WA

Full-time - Senior

Seattle, WA

Computer and Electronic Product Manufacturing

About the position

As a Machine Learning Engineer at Apple, you will be part of an applied research and engineering team focused on developing real-time on-device Language, Computer Vision, and Machine Perception technologies. This role involves optimizing and deploying Apple Intelligence LLM and diffusion models, contributing to the full ML model lifecycle, and collaborating with various teams to enhance user experiences. You will lead the execution of model compression and distillation techniques, ensuring high performance of large models on devices while also having the opportunity to publish research at top ML conferences.

Responsibilities

• Implement and deliver optimization techniques for large language and diffusion models on devices.
• Define and lead the execution of model compression and distillation.
• Integrate optimized models into the full Apple Intelligence user experiences.
• Collaborate with diverse organizations within Apple to enhance ML model lifecycle.
• Drive the development of on-device Apple Intelligence LLM and diffusion model developments.

Requirements

• Software engineering skills in Python.
• Experience in developing large computer vision and machine learning models, particularly on hardware-aware model optimizations.
• BS degree and a minimum of 10 years relevant industry experience.

Nice-to-haves

• Familiarity with model compression algorithms including quantization, pruning, and distillation.
• Experience optimizing large diffusion models or language models.
• MS or PhD degree in Computer Science or equivalent industry research experience.
• Experience with hardware architecture and software & hardware co-design.
• Leadership experience in driving large-scale projects in the industry.
• Strong communication skills and a phenomenal work ethic.

Benefits

• Comprehensive medical and dental coverage
• Retirement benefits
• Discounted products and free services
• Reimbursement for certain educational expenses including tuition
• Discretionary bonuses or commission payments
• Relocation assistance