Large Machine Learning Model Optimization Software Engineer

Apple • Seattle, WA

66d

About The Position

Our team is an applied research and engineering team responsible for developing real-time on-device Language, Computer Vision, and Machine Perception technologies across Apple products. We focus on technology research and development to deliver Apple quality, state-of-the-art experiences. Our team prides itself on innovating through the full stack, and partnering with HW, SW and ML teams to influence the sensor and silicon roadmap that brings our vision to life. We are directly responsible for the on-device optimization and deployment of the Apple Intelligence LLM and diffusion models. As a Machine Learning Engineer, you will have the opportunity to be at the forefront of technological advancements and contribute to the successful shipping and delivery of Apple intelligence. You will be responsible for implementing and delivering various optimization techniques that improve the performance of large language and diffusion models on devices. Additionally, you will collaborate with a diverse range of organizations within Apple. Your innovations will significantly impact the entire ML model lifecycle of Apple intelligence.

Requirements

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

Nice To Haves

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

Responsibilities

• Drive the development of the on-device Apple Intelligence LLM and diffusion model developments.
• Define and lead the execution of model compression, distillation, and integration into the full Apple Intelligence user experiences.
• Implement and deliver various optimization techniques that improve the performance of large language and diffusion models on devices.
• Collaborate with a diverse range of organizations within Apple.
• Encourage publishing novel research at top ML conferences.

What This Job Offers

• Number of Employees: 5,001-10,000 employees