Senior Machine Learning Engineer, Apple Search & Knowledge Platforms

The AI Search u0026 Knowledge Platforms team builds amazing products and services for Apples 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. nnAs part of this group you will be doing large scale machine learning and deep learning research and development to improve Open Domain Question Answering (using both structured knowledge graph data and unstructured web data) and Summarization as well as developing fundamental building blocks needed for Artificial Intelligence. This involves developing sophisticated machine learning and large language models (LLMs) to understand user queries retrieve and rank relevant documents across multiple sources and synthesize information across documents to provide user with a direct answer that best satisfies their intent and information seeking needs. Additionally you will research and develop the state-of-the-art LLMs for summarizing personal data such as emails messages and notifications. nnYou will also work with researchers and data scientists to develop fine-tune and evaluate domain specific Large Language Models for various tasks and applications in Apples AI powered products and conduct applied research to transfer the cutting edge research in generative AI to production ready technologies.

As a member of our fast-paced group youll have the unique and rewarding opportunity to shape upcoming products from Apple. We are looking for highly motivated machine learning engineers and researchers having strong machine learning and deep learning fundamentals with hands-on experience in fine-tuning deep learning and large language models.

Conduct research and development on state-of-the-art deep learning and large language models for various tasks and applications in Apples AI-powered productsnDeveloping fine-tuning and evaluating domain-specific Large Language Models for various NLP tasks including summarization question answering search relevance/ranking entity linking and query understanding problemsnConducting applied research to transfer the cutting edge research in generative AI to production ready technologiesnUnderstanding product requirements translate them into modeling tasks and engineering tasksnStay up to date with the latest advancements and research in deep learning and large language models

2 years of experience working with Deep learning or LLM model development for various NLP tasks and RAG applications including prompt engineering training data collection and generation model fine-tuning and model working with Python and at least one of the deep learning frameworks such as TensorFlow PyTorch or in Computer Science Artificial Intelligence Machine Learning or a related field.

PhD in Computer Science Artificial Intelligence Machine Learning or a related field.n4 years of experience with large-scale model training optimization and deploymentnOne or more scientific publications in various conferences and journalsnOutstanding communication and interpersonal skills with ability to work with cross-functional teams.n1 year of experience in various state-of-the-art techniques related to LLM fine-tuning in 1 or more of the following areas:nSupervised Fine-tuning (SFT) with Rejection SamplingnPreference-based fine-tuning techniques (e.g RLHF Reward model DPO PPO GRPO etc.)nParameter efficient fine-tuning techniques (e.g LoRA)nHallucination reduction and factual accuracy improvementsnDesigning and implementing safety guardrails