Staff Machine Learning Engineer, Search & Knowledge Platform

Apple Inc

Quick summary

Work type
On-site
Location
Seattle, WA
Salary
$171,600–$302,200 / yr
Posted
36 days ago

Market check

Salary context

Above market

How this pay compares to similar roles

Similar $214k
This role $237k
$156k most similar roles pay here $318k

This role pays more than 66% of similar roles. Most pay $178,689–$249,750 — the shaded band above. At the midpoint, this role pays about $237k versus about $214k for comparable roles.

Based on 239 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 638 open roles on FindRole.

Listed pay typically runs $171,600–$272,100 across 505 roles with salary data.

Most-posted roles

View all roles at Apple Inc

At a glance

TL;DR · Staff Machine Learning Engineer, Search & Knowledge Platform

As a Staff Machine Learning Engineer on Apple’s Search & Knowledge Platform team, you will play a pivotal role in shaping the next generation of search technologies across Siri, Safari, Spotlight, and other key products. Your responsibilities include translating product requirements into modeling tasks, analyzing ranking and relevance issues, and utilizing frameworks like PyTorch, TensorFlow, or JAX to develop sophisticated ML models for retrieval, ranking, and query understanding. You will work with petabytes of data, integrating information from various sources to enhance user experiences through cutting-edge techniques such as LLMs and RAG. This role demands expertise in deep learning, strong software engineering skills, and experience in shipping search and Q&A technologies at scale.

What you'll do

  • Translate product requirements into modeling and engineering tasks.
  • Analyze search ranking and relevance issues to identify improvement opportunities.
  • Utilize PyTorch, TensorFlow, or JAX for training and deploying deep learning models.
  • Build ML models for retrieval, relevance ranking, and query understanding.
  • Leverage and improve upon the latest deep learning techniques like LLMs and RAG.

What we're looking for

  • MS in Computer Science or related field with 10+ years of machine learning experience
  • Extensive experience shipping Search and Q&A technologies and ML systems
  • Proficiency in PyTorch, TensorFlow, JAX for training and deploying deep learning models
  • Strong programming skills in C++, Python, Scala, and Go
  • Experience delivering tooling to evaluate individual components and end-to-end quality
  • Analytical skills to improve search relevance and answer accuracy systematically
  • Excellent communication and interpersonal skills for collaboration

More like this

Similar roles

Sr Staff Machine Learning Engineer, ML Platform

Apple Inc

Cupertino, CA 9 days ago $212,000$386,300
Python TensorFlow PyTorch Kubernetes Docker CI/CD Prometheus Grafana PostgreSQL AWS Azure MLOps Federated Learning Differential Privacy LLMs Agentic AI

Staff Machine Learning Engineer

Apple Inc

Cupertino, CA 43 days ago $212,000$318,400
Python PyTorch JAX TensorFlow Spark Daft Rust Java Go Kubernetes Docker CI/CD Parquet Iceberg Delta Lance Ray Data NVIDIA DALI WebDataset Mosaic StreamingDataset Arrow DataHub OpenLineage Unity Catalog Polars DuckDB

Sr Staff Machine Learning Engineer, ML Platform

Apple Inc

New York City, NY 9 days ago $212,000$386,300
Python TensorFlow PyTorch Kubernetes Docker CI/CD Prometheus Grafana PostgreSQL AWS Azure Google Cloud Platform Federated Learning Differential Privacy Transformers LLMs DNNs Model Pruning Compression Quantization Distillation Agentic AI

Staff Machine Learning Engineer, Search Ranking

Snap Inc.

Santa Monica, CA 2 days ago $229,000$343,000
Python TensorFlow PyTorch Spark Flink Beam Java Scala C++ JAX A/B testing learning-to-rank LambdaMART neural ranking models transformer-based rankers large-scale data processing ML infrastructure online experimentation model monitoring feature pipelines training infrastructure serving systems multi-objective optimization LLMs foundation models semantic search natural language understanding retrieval-augmented generation