Senior Machine Learning Engineer, Search & Platform

Staff/Senior Machine Learning Engineer, Search & Knowledge Platform

Seattle, Washington, United States Software and Services

Apple is where individual imaginations come together, committing to the values that lead to great work. Every new product we build, service we create, or Apple Store experience we deliver is the result of us strengthening each other’s ideas. That happens because every one of us believes that we can make something wonderful and share it with the world, changing lives for the better!

Here, you’ll do more than join something — you’ll add something. The Information Intelligence team is redefining how billions of people use their devices to get information. We are an Applied ML team pushing the limits of question answering, 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.

As a Staff/Senior Machine Learning Engineer, you play a critical role in developing world-class Search and Q&A experiences for Apple customers with cutting-edge search technologies and large language models.

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 scientists with a focus in NLP to experienced distributed systems engineers. As such, we are looking for candidates with in-depth understanding of machine learning fundamentals, applied machine learning experience, and strong software engineering skills.

Our team is responsible for delivering next-generation Search and Question Answering systems across Apple products including Siri, Safari, Spotlight, and more. This is your chance to shape how people get information by leveraging your Search and applied machine learning expertise along with robust software engineering skills. You will collaborate with outstanding Search and AI engineers on large scale machine learning to improve Query Understanding, Retrieval, and Ranking, developing fundamental building blocks needed for AI powered experiences such as fine-tuning and reinforcement learning.

This involves pushing the boundaries on document retrieval and ranking, developing sophisticated machine learning models, using embeddings and deep learning to understand the quality of matches. It also includes online learning to react quickly to change and natural language processing to understand queries. You will work with petabytes of data and combine information from multiple structured and unstructured sources to provide the best results and accurate answers to satisfy users' information-seeking needs.

As part of our team, you will be leveraging and improving upon the latest deep learning techniques, such as LLM and RAG, in order to understand queries and user intents, rank documents, and find useful answers to users’ questions. Our team is responsible for training, fine-tuning and deploying these models at scale, using the latest advances for online inference optimization.

Following is the primary list of responsibilities for an ML engineer in the team:

• Understanding product requirements then translating them into modeling and engineering tasks
• Analyzing search ranking and relevance requirements, issues, and opportunities
• Utilizing PyTorch, Tensor Flow, or JAX for training and deploying deep learning models
• Building ML models for retrieval, relevance ranking, and query understanding

• MS in Computer Science or related field
• 10+ years of work experience in machine learning, deep learning or related field
• 10+ years experience in shipping Search and Q&A technologies and ML systems
• Excellent programming skills in mainstream programming languages such as C++, Python, Scala, and Go
• Experience delivering tooling and frameworks to evaluate individual components and end-to-end quality
• Strong analytical skills to systematically identify opportunities to improve search relevance and answer accuracy
• Strong written and verbal communication with the…