Senior Staff Machine Learning Engineer, Apple Search & Knowledge Platforms

In this role, you'll drive E2E R&D and engineering to generate high-quality summaries and experiences for Apple users. This includes on-device LLM models for personal content summarization across 1P and 3P apps, and powerful summarization models on Apple’s Private Cloud Compute servers. In addition, improve the summarization models’ quality for world knowledge-seeking questions and Safari pages to provide accurate answers and highlight web page gists in real-time. Lead the team to develop SOTA LLM-based generative models, groundedness models, and safety models for accurate, grounded, concise, and safe summaries. Develop sophisticated on-device and on-server software frameworks for context integration fast and cost-efficient LLM-based model inference. Integrate the Apple ecosystem with Apple’s LLM infrastructure and generative models to deliver delightful user experiences. Devise the product vision and strategy and execute the plan to deliver the highest quality end-user experience. Collaborate with various organizational partners to profoundly impact billions of Apple users worldwide.

8+ years of experience in leading engineering/applied research/ML experiences in natural language processing, SOTA generative AI models
Proven record of consistent delivery of technology/products across the full Machine Learning life cycle
MS or Ph.D. in Computer Science, Machine Learning, information retrieval, data mining, or a related field

Strong background and experience in Machine Learning, NLP, and RAG.
Strong engineering and R&D experience in LLM post-training, advanced RL-based methods to improve LLM models’ safety and quality using RLHF/RLAIF, reward model, advanced RL policy optimization algorithms, cutting-edge hallucination reduction methods, and their engineering implementation, hands-on experience to develop and ship RL based models with high availability, low latency, robustness, and stability.
Exceptional verbal and written communication skills to lead
Excellent product vision and sound business acumen. Ability to manage long-term strategy and short-term deliverables.
Strong engineering leadership and fundamentals.

Apple is where individual imaginations gather together, contributing to the values that lead to great work. Every The AI, Search & Knowledge Platforms team builds amazing products and services for Apple's customers while serving as a foundational partner to teams across Apple. The team delivers world-class AI, search, and knowledge systems powering Siri, Apple Intelligence, Safari, and iMessage, and operates the foundational platforms and infrastructure that keep these intelligent experiences running at hyperscale.

You will lead the strong team of MLE, SWE, and data engineers responsible for delivering efficient and effective Generative AI models to build and improve the summarization capabilities across different data types.

At Apple, base pay is one part of our total compensation package and is determined within a range. This provides the opportunity to progress as you grow and develop within a role. The base pay range for this role is between $201,300 and $367,400, and your base pay will depend on your skills, qualifications, experience, and location.

Apple employees also have the opportunity to become an Apple shareholder through participation in Apple’s discretionary employee stock programs. Apple employees are eligible for discretionary restricted stock unit awards, and can purchase Apple stock at a discount if voluntarily participating in Apple’s Employee Stock Purchase Plan. You’ll also receive benefits including: Comprehensive medical and dental coverage, retirement benefits, a range of discounted products and free services, and for formal education related to advancing your career at Apple, reimbursement for certain educational expenses — including tuition. Additionally, this role might be eligible for discretionary bonuses or commission payments as well as relocation. Learn more about Apple Benefits

Note: Apple benefit, compensation and employee stock programs are subject to eligibility requirements and other terms of the applicable plan or program.

Apple is an equal opportunity employer that is committed to inclusion and diversity. We seek to promote equal opportunity for all applicants without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, disability, Veteran status, or other legally protected characteristics. Learn more about your EEO rights as an applicant

At Apple, we believe accessibility is a fundamental human right. You’ll find that idea reflected in everything here — in our culture, our benefits and our digital tools. By welcoming as many perspectives as possible, we help you build a career where you feel like you belong.

Learn about accessibility in Apple’s workplace

Learn about reasonable accommodations for job applicants

Apple accepts applications to this posting on an ongoing basis.

At Apple, base pay is one part of our total compensation package and is determined within a range. This provides the opportunity to progress as you grow and develop within a role. The base pay range for this role is between $212,000 and $386,300, and your base pay will depend on your skills, qualifications, experience, and location.

Apple employees also have the opportunity to become an Apple shareholder through participation in Apple’s discretionary employee stock programs. Apple employees are eligible for discretionary restricted stock unit awards, and can purchase Apple stock at a discount if voluntarily participating in Apple’s Employee Stock Purchase Plan. You’ll also receive benefits including: Comprehensive medical and dental coverage, retirement benefits, a range of discounted products and free services, and for formal education related to advancing your career at Apple, reimbursement for certain educational expenses — including tuition. Additionally, this role might be eligible for discretionary bonuses or commission payments as well as relocation. Learn more about Apple Benefits

Note: Apple benefit, compensation and employee stock programs are subject to eligibility requirements and other terms of the applicable plan or program.

Apple is an equal opportunity employer that is committed to inclusion and diversity. We seek to promote equal opportunity for all applicants without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, disability, Veteran status, or other legally protected characteristics. Learn more about your EEO rights as an applicant

At Apple, we believe accessibility is a fundamental human right. You’ll find that idea reflected in everything here — in our culture, our benefits and our digital tools. By welcoming as many perspectives as possible, we help you build a career where you feel like you belong.

Learn about accessibility in Apple’s workplace

Learn about reasonable accommodations for job applicants

Apple accepts applications to this posting on an ongoing basis.