Senior Machine Learning Engineer, Search & Knowledge Platforms

Do you want to make Siri and Apple products smarter for our users? The Answers, Knowledge \u0026 Information team is redefining how hundreds of millions of people use their devices to get information. We are an Applied ML team pushing the limits of apple intelligence, assistant response ranking, and search technologies, while also responsible for a production service. We are part of a wider effort to power information across a variety of Apple products - including Siri, Spotlight, Safari, Messages, Lookup, and more. In our team, you will be leveraging and improving upon the latest LLM/ML techniques in order to understand queries, rank user intents, rank documents, and generate answers to users’ questions. Our team is responsible for training and deploying these models at scale, using the latest advances for online inference optimization.

As a member of our fast-paced group, you’ll have the unique and rewarding opportunity to shape upcoming products from Apple. Our team includes a diversity of backgrounds from applied machine learning engineers with a focus on ML and LLM to experienced distributed systems engineers. As such, we are looking for candidates with applied machine learning experience and strong software engineering skills.

Build world class large language model at scale to power question and answering system for Siri\nCraft and develop core software infrastructure to support natural language understanding system in production\nDesign and implement the low latency and high reliability runtime tech stacks for global search\nDesign and develop data pipeline and infrastructure for large scale of data processing\nEvaluate and benchmark Apple natural language processing models, and deploy them to serve billions of Apple users

8+ years of industry related experience, working in collaborate environments\nUtilizing PyTorch, TensorFlow, and JAX for training and deploying deep learning models\nUnderstanding product requirements and then translating them into modeling tasks and engineering tasks\nBuilding models for search relevance ranking, query understanding, and summarization\nBS or MS in Computer Science or related field

PhD in Computer Science, Artificial Intelligence, Machine Learning, Information Retrieval, Data Science or related field\nExperience in RAG, LLM Reasoning, and LLM Agent\nExperience working in a complex organization with multiple collaborators, strong track record in scaling and launching projects in this setting