Staff Machine Learning Engineer, Answers, Knowledge & Information

Apple Inc

Confirmed live yesterday High trust

Quick summary

Work type
On-site
Location
Santa Clara, CA
Salary
$184,700–$324,800 / yr
Posted
1 day 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,800–$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 2305 open roles on FindRole.

Listed pay typically runs $175,000–$280,000 across 1873 roles with salary data.

Most-posted roles

View all roles at Apple Inc

At a glance

TL;DR · Staff Machine Learning Engineer, Answers, Knowledge & Information

AIML - Staff Machine Learning Engineer, Answers, Knowledge & Information (AKI) joins the Siri and Information Intelligence team to develop LLMs and NLP models focused on personal domain questions and user productivity. The role involves researching, designing, implementing, and evaluating models to enhance performance for Personal Q&A features while maintaining strict privacy standards. Key responsibilities include extending LLM models through fine-tuning and RLXF methodologies, building high-quality data pipelines, establishing evaluation metrics, and defining long-term technical visions for the product roadmap. The position requires expertise in Python, Go, or C/C++ along with experience in frameworks like PyTorch, JAX, TensorFlow, and XGBoost. Candidates will work on complex problems involving search relevance, ranking, information retrieval, and response generation to power intelligent search experiences across various integrated products.

What does a Machine Learning Engineer earn in California?

Median $246150 from 187 postings across 29 companies.

See salary data

What you'll do

  • Research, design, and implement models to enhance quality and performance for Personal Q&A functions.
  • Implement and extend LLM models using fine-tuning and RLHF methodologies.
  • Develop high-quality data pipelines to support model training and refinement.
  • Establish target quality metrics and evaluation sets to track and improve system performance.
  • Design novel experiments to validate hypotheses and support team-wide decision making.
  • Define product requirements and priorities in collaboration with partner teams.
  • Develop a long-term technical vision for Personal Q&A quality and integrate solutions into the roadmap.

What we're looking for

  • Bachelor's degree in Computer Science.
  • 8+ years of industry experience in Machine Learning and NLP at scale.
  • Strong software engineering skills in Python, Go, or C/C++.
  • Experience using ML frameworks such as PyTorch, JAX, TensorFlow, or XGBoost.
  • Strong communication skills.
  • In-depth knowledge of training and applying Deep learning models and LLMs for production systems (preferred).
  • Extensive experience building production ML systems in search, recommendation, or information retrieval (preferred).
  • Advanced degree in Computer Science, Statistics, or a related field (preferred).

More like this

Similar roles

Machine Learning Engineer, Answers Knowledge and Information

Apple Inc

Cupertino, CA 1 day ago $150,400$277,600
Large Language Models LLMs Natural Language Processing Python PyTorch TensorFlow JAX Deep Learning Generative AI RAG Prompt Engineering RLHF DPO PPO LoRA Supervised Fine-tuning Recommendation Systems
10+ yrs exp

Machine Learning Researcher, Data and ML Innovation

Apple Inc

Seattle, WA 1 day ago $142,300$263,300
Machine Learning Deep Learning Natural Language Processing Large Language Models Generative AI Python Go Java C++ Spark Hadoop MapReduce Hive Impala Information Retrieval Data Mining Prompt Engineering Word Embeddings Distributed Data Processing

Staff Senior Machine Learning Engineer, Search & Knowledge Platform

Apple Inc

Seattle, WA 20 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 49 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

Machine Learning Engineer

Apple Inc

Cupertino, CA 155 days ago $184,700$324,800
C++ Python GPU Programming Machine Learning Swift Numerical Computing Distributed Training Software Engineering Low-level Programming MLX Apple Silicon