Large Machine Learning Model Optimization Engineer

Apple is a technology company known for developing innovative products and services in consumer electronics, software, and online services.
$207,800 - $378,700
Machine Learning
Principal Software Engineer
In-Person
5,000+ Employees
20+ years of experience
AI · Consumer · Enterprise SaaS

Description For Large Machine Learning Model Optimization Engineer

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.

We're looking for strong ML software engineers/leaders to drive the development of the on-device Apple Intelligence LLM and diffusion model developments. This includes defining and leading the execution of model compression, distillation, and integrating to the full Apple Intelligence user experiences. We expect you to have strong, efficient ML model development experiences and a passion for shipping machine learning models on device. We also encourage publishing novel research at top ML conferences.

Last updated 15 days ago

Responsibilities For Large Machine Learning Model Optimization Engineer

  • Drive the development of on-device Apple Intelligence LLM and diffusion model developments
  • Define and lead the execution of model compression, distillation, and integration to full Apple Intelligence user experiences
  • Implement and deliver various optimization techniques to improve performance of large language and diffusion models on devices
  • Collaborate with diverse organizations within Apple
  • Contribute to the successful shipping and delivery of Apple intelligence
  • Publish novel research at top ML conferences

Requirements For Large Machine Learning Model Optimization Engineer

Python
  • Software engineering skills in Python
  • Experience in developing large computer vision and machine learning models, particularly on the hardware-aware model optimizations
  • BS and a minimum of 20 years relevant industry experience
  • Familiar with model compression algorithms including quantization, pruning, distillations, and experience on 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

Benefits For Large Machine Learning Model Optimization Engineer

Medical Insurance
Dental Insurance
401k
Equity
Education Budget
  • Comprehensive medical and dental coverage
  • Retirement benefits
  • Discounted products and free services
  • Reimbursement for certain educational expenses
  • Opportunity to become an Apple shareholder through discretionary employee stock programs
  • Employee Stock Purchase Plan
  • Potential eligibility for discretionary bonuses or commission payments
  • Potential eligibility for relocation assistance

Interested in this job?

Jobs Related To Apple Large Machine Learning Model Optimization Engineer

AIML- Software Development Engineer, Machine Learning Platforms & Infrastructure

Lead Systems Engineer / Systems Architect role at Apple, focusing on Machine Learning infrastructure and platforms, requiring 15+ years of experience.

Research Scientist 4 - Content and Studio

Netflix seeks Research Scientist 4 for Content and Studio team, focusing on Computer Vision to create innovative Visual Promotional Media Assets.

Principal Machine Learning Engineer

Principal Machine Learning Engineer at Sezzle: Lead AI/ML development for fintech platform, focusing on scalable solutions and team leadership.

Principal ML Engineer - Scenario Technology and Coverage

Principal ML Engineer leading scenario technology for autonomous vehicles at Wayve

Senior Staff Machine Learning Engineer, Trust

Senior Staff Machine Learning Engineer for Airbnb's Trust team, building cutting-edge ML models to protect the platform and community.