Staff Senior Machine Learning Engineer, AI, Search & Knowledge Platforms

Apple Inc

Confirmed live yesterday High trust

Quick summary

Work type
On-site
Location
San Francisco, CA
Salary
$184,700–$324,800 / yr
Posted
58 days ago
Freshness
Confirmed live yesterday

Market check

Salary context

Above market

How this pay compares to similar roles

Similar $229k
This role $255k
$168k most similar roles pay here $342k

This role pays more than 76% of similar roles. Most pay $202,441–$254,750 — the shaded band above. At the midpoint, this role pays about $255k versus about $229k for comparable roles.

Based on 240 similar postings.

Employer

About Apple Inc

Apple Inc. is a multinational technology company known for designing and manufacturing consumer electronics, software, and online services, including the iPhone, Mac, iPad, and App Store. Industry: Consumer Electronics & Software

Apple Inc currently has 1984 open roles on FindRole.

Listed pay typically runs $175,000–$277,600 across 1590 roles with salary data.

Most-posted roles

View all roles at Apple Inc

At a glance

TL;DR · Staff Senior Machine Learning Engineer, AI, Search & Knowledge Platforms

Staff / Sr. Machine Learning Engineer, AI, Search & Knowledge Platforms joins the Information Intelligence Infrastructure team to build and optimize large-scale extraction and enrichment pipelines. This role involves transforming raw web content into structured knowledge for products like Siri and Spotlight by designing scalable algorithms and advanced deduplication techniques. The engineer will develop ML classifiers for data quality, manage model serving and batch workloads, and create high-fidelity training corpora for foundation models to improve reasoning and grounding. Key technical requirements include proficiency in Golang, Java, Scala, or Python, along with experience in Spark, distributed systems, and cloud-native environments like AWS. The role specifically addresses the challenge of processing trillions of records to maintain a unified Knowledge Graph while utilizing large language models, NLP, entity linking, and inference frameworks such as TensorRT-LLM or vLLM.

What does a Machine Learning Engineer earn in California?

Median $246394 from 172 postings across 28 companies.

See salary data

What you'll do

  • Build and optimize large-scale extraction pipelines to transform raw web data into structured knowledge.
  • Design pipelines using LLMs, advanced NLP, and entity linking to deduplicate and contextualize information.
  • Optimize infrastructure for cost, throughput, and reliability across model serving and batch ML workloads.
  • Develop machine learning classifiers for data quality, deduplication, and content filtering across billions of records.
  • Improve extraction quality to create high-fidelity training corpora for Apple's foundation models.
  • Maintain the Knowledge Graph by integrating diverse data sources into a unified representation.

What we're looking for

  • Bachelor's degree or higher in Computer Science or a related technical field.
  • 4+ years of experience in fine-tuning LLMs, building large-scale extraction pipelines, or optimizing ML infrastructure.
  • Experience taking machine learning models to production, including training and evaluation.
  • Proficiency in Golang, Java, Scala, or Python.
  • Strong background in computer science fundamentals, including algorithms, data structures, and distributed systems.
  • Experience working in cloud-native environments such as AWS.
  • Experience with large-scale data processing pipelines using Spark or similar distributed frameworks.
  • Familiarity with NLP, information extraction, web data systems, and ML inference frameworks like TensorRT-LLM or vLLM.

More like this

Similar roles

Staff Senior Machine Learning Engineer, Search & Knowledge Platforms

Apple Inc

Seattle, WA 17 days ago $175,000$308,500
Machine Learning Python Go Large Language Models (LLMs) Natural Language Processing (NLP) Information Retrieval Retrieval-Augmented Generation (RAG) Prompt Engineering Data Science A/B Testing Statistical Hypothesis Testing Data Pipelines Distributed Inference Model Quantization Search Ranking
6+ yrs exp

Staff Senior Machine Learning Engineer, Search & Knowledge Platforms

Apple Inc

Santa Clara, CA 17 days ago $184,700$324,800
Machine Learning Python Go LLMs Prompt Engineering Information Retrieval Natural Language Processing RAG Data Pipelines A/B Testing Statistical Hypothesis Testing Search Ranking Model Quantization Distributed Inference Data Science
6+ yrs exp

Staff Senior Machine Learning Engineer, Search & Knowledge Platform

Apple Inc

Seattle, WA 13 days ago $175,000$308,500
Machine Learning Deep Learning LLM RAG NLP PyTorch TensorFlow JAX Python C++ Scala Go Information Retrieval Question Answering Retrieval Ranking Query Understanding Fine-tuning Reinforcement Learning Inference Optimization
10+ yrs exp

Staff Senior Machine Learning Engineer, Search & Knowledge Platform

Apple Inc

Seattle, WA 42 days ago $175,000$308,500
Machine Learning Deep Learning LLM RAG NLP PyTorch TensorFlow JAX Python C++ Go Scala Information Retrieval Question Answering Retrieval Ranking Query Understanding Online Inference Optimization Fine-tuning Reinforcement Learning
10+ yrs exp