Senior Machine Learning Engineer, Search & Knowledge Platforms

Apple Inc.; Seattle, Northern (WA, KY); Hybrid; USD 175,000 - 309,000; Full time

Job summary

Apple Inc. in Seattle is seeking a Senior Machine Learning Engineer for the Search & Knowledge Platforms to build and scale large language models powering Siri and related information services.

You will design core software infrastructure for production NL understanding, optimize for low latency, and deploy models that serve billions of Apple users.

Qualifications

• 8+ years of industry related experience
• Experience with PyTorch, TensorFlow, and JAX
• Translating product requirements into modeling tasks and engineering tasks
• Building models for search relevance ranking, query understanding, and summarization
• BS or MS in Computer Science or related field

Responsibilities

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

Skills

Large language models

PyTorch

TensorFlow

JAX

Query understanding

Search relevance ranking

Education

BS/MS in Computer Science or related field

PhD in CS/AI/ML/IR/DS or related field

Tools

PyTorch

TensorFlow

JAX

Job description

Senior Machine Learning Engineer, Search & Knowledge Platforms

Seattle, Washington, United States Machine Learning and AI

Do you want to make Siri and Apple products smarter for our users? The Answers, Knowledge & 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.

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 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.

Responsibilities

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

Minimum Qualifications

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

Preferred Qualifications

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

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.