Staff/Senior Machine Learning Engineer, Search & Knowledge Platform

Apple Inc.; Seattle (WA); On-site; USD 175,000 - 309,000; Full time

Benefits offered by this job

Employee stock programs

Educational reimbursement

Relocation assistance

Medical and dental coverage

Job summary

Apple Inc. in Seattle, Washington is seeking a Staff/Senior Machine Learning Engineer for the Information Intelligence team to build world-class Search and Q&A experiences across Apple products.

You will work with cutting-edge ML, fine-tuning and online inference at scale, and collaborate with engineers across Siri, Safari, Spotlight, and more.

Qualifications

• MS in Computer Science or related field.
• 10+ years of work experience in ML, deep learning or related field.
• 10+ years shipping Search and Q&A technologies and ML systems.
• Excellent programming skills in C++, Python, Scala, and Go.
• Experience delivering tooling to evaluate components and end-to-end quality.
• Strong analytical skills to improve search relevance and answer accuracy.
• Strong written and verbal communication.
• Excellent interpersonal skills and teamwork.
• Passion for building phenomenal products and curiosity to learn.

Responsibilities

• Understanding product requirements and translating to modeling tasks.
• Analyzing search ranking and relevance requirements and opportunities.
• Utilizing PyTorch, TensorFlow, or JAX for training and deploying models.
• Building ML models for retrieval, relevance ranking, and query understanding.

Skills

C++

Python

Scala

Go

PyTorch

TensorFlow

JAX

Education

MS in Computer Science

PhD in CS/AI

Tools

PyTorch

TensorFlow

JAX

Job description

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.

Description

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:

Responsibilities

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

Minimum Qualifications

• 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 ability to articulate complex topics
• Excellent interpersonal skills and teamwork; demonstrated ability to connect and collaborate with others
• Passion for building phenomenal products and curiosity to learn

Preferred Qualifications

• PhD in Computer Science, Artificial Intelligence, Machine Learning, Information Retrieval, Data Science or related field
• Strong industry background and experience in search and related technologies (LLMs, Machine Learning, NLP, Information Retrieval, Question Answering)
• Strong and validated experience of ML development and production systems
• Experience working with foundation models and LLMs

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 $175,000 and $308,500, 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.