Drimva

Apple

Staff Machine Learning Engineer – Ads Predictions

$216K–$325K

Cupertino, California · Staff · Full-time · Posted Aug 12, 2026

Experience
8+ years; no alternatives stated.
Education
Bachelor's, or equivalent experience, in Computer Science, Machine Learning, Artificial Intelligence, Information Retrieval, or a related field.

About the job

Build and scale machine-learning models and retrieval systems that predict ad interactions and improve advertising outcomes across Apple’s App Store and News platforms.

What you'll do

  • Design and improve models predicting user interaction, click-through rate, and conversion rate; develop retrieval algorithms.
  • Work with large-scale datasets and cross-functional teams to productionize models and identify useful signals.
  • Design and run experiments to validate model architectures and learning strategies.

What they ask for

  • 8+ years applying machine learning and statistical modeling at scale.
  • Deep experience with neural network architectures and training pipelines using TensorFlow or PyTorch; practical understanding of reinforcement learning and bandit-based optimization.
  • Experience with high-volume data pipelines, A/B testing, and performance measurement; proficiency in Python and familiarity with SQL, Scala, or Java.
  • Bachelor's, or equivalent experience, in Computer Science, Machine Learning, Artificial Intelligence, Information Retrieval, or a related field.

Nice to have

  • MS or PhD, or equivalent experience, in a related field.
  • Foundation in information retrieval, including query-document matching, embedding-based ranking, and learning-to-rank algorithms.

Pay

$216,200 to $324,800 USD base salary

Tools

TensorFlow, PyTorch, Python, SQL, Scala, Java

AI focus

Develop ad prediction, retrieval, and ranking systems using machine learning, deep learning, reinforcement learning, contextual bandits, multi-task learning, and LLM-based ranking signals.

Model-written summary · the original posting is authoritative

First seen Oct 15, 2026 · Last verified Nov 5, 2026 (UTC)