Staff Machine Learning Engineer, Retrieval

Reddit

Confirmed live yesterday High trust
Remote

Quick summary

Work type
Remote
Location
Remote
Salary
$230,000–$322,000 / yr
Posted
9 days ago
Freshness
Confirmed live yesterday

Market check

Salary context

Above market

How this pay compares to similar roles

Similar $231k
This role $276k
$168k most similar roles pay here $338k

This role pays more than 90% of similar roles. Most pay $207,800–$254,750 — the shaded band above. At the midpoint, this role pays about $276k versus about $231k for comparable roles.

Based on 240 similar postings.

Employer

About Reddit

Reddit is a social news aggregation and discussion platform where users share content, vote on posts, and engage in community conversations across thousands of interest-based forums called subreddits.

Reddit currently has 79 open roles on FindRole.

Listed pay typically runs $217,000–$303,000 across 79 roles with salary data.

Most-posted roles

View all roles at Reddit

At a glance

TL;DR · Staff Machine Learning Engineer, Retrieval

As a Staff Machine Learning Engineer on the Ads Retrieval ML team, you will provide technical leadership to design and evolve retrieval models that identify relevant advertising candidates for users. You will define multi-year roadmaps, develop candidate-generation models across various surfaces, and manage the full lifecycle from data preparation to production launch. The role involves building systems using representation learning, two-tower architectures, sequence models, graph-based methods, and approximate nearest-neighbor search. You will work with large-scale behavioral and contextual datasets while utilizing frameworks like TensorFlow or PyTorch. Your core responsibilities include optimizing retrieval metrics such as recall and precision to improve advertiser outcomes and user experience. This role solves the critical problem of determining which campaigns are eligible for competition within a complex, high-volume advertising funnel involving diverse objectives and geographies.

What does a Machine Learning Engineer earn in Remote?

Median $235875 from 52 postings across 19 companies.

See salary data

What you'll do

  • Define the technical direction and multi-year roadmap for ads retrieval modeling across various stakeholders.
  • Design, develop, and launch candidate-generation and retrieval models for campaigns and ads on multiple surfaces.
  • Implement advanced deep learning techniques including two-tower architectures, representation learning, and sequence models.
  • Optimize the retrieval stack through improvements in objective design, sampling strategies, feature engineering, and embedding generation.
  • Manage approximate nearest-neighbor systems while balancing trade-offs between recall, latency, cost, and diversity.
  • Establish evaluation frameworks that connect offline retrieval metrics to downstream business outcomes and user experience.
  • Lead offline analysis and online experiments to interpret results and drive the next modeling iterations.
  • Mentor machine learning engineers and provide technical leadership on complex projects and production standards.

What we're looking for

  • 7+ years of industry experience building and shipping applied ML products.
  • Deep experience in information retrieval, candidate generation, recommender systems, or ranking.
  • Strong understanding of retrieval modeling including DNNs, embeddings, two-tower models, and approximate nearest-neighbor search.
  • Deep experience training, evaluating, debugging, and deploying deep learning models using TensorFlow or PyTorch.
  • Demonstrated ownership of ML projects from problem framing through production launch and iteration.
  • Strong software engineering fundamentals and ability to write reliable, maintainable production code.
  • Technical leadership experience including setting direction, leading complex projects, and mentoring engineers.
  • Experience with ads retrieval, search relevance, or marketplace optimization (preferred); publications or patents in applied ML (preferred).

More like this

Similar roles

Staff Machine Learning Engineer, Applied AI

Uber

San Francisco, CA +2 70 days ago
Generative AI LLM PyTorch Transformers Distributed Training Retrieval Systems Ranking Embedding Architectures Computer Vision Geospatial AI Search Conversational AI Deep Learning
8+ yrs exp

Staff Machine Learning Engineer, Search Ranking

Snap Inc.

Bellevue, WA +4 114 days ago $229,000–$343,000
Machine Learning Python C++ Java Scala Spark Flink Beam TensorFlow PyTorch JAX LLM Retrieval-Augmented Generation A/B Testing Search Ranking Embedding Neural Ranking Multi-task Learning Vector Search
8+ yrs exp

Senior Machine Learning Engineer, Ads Predictions

Apple Inc

New York, NY 161 days ago $216,200–$324,800
Machine Learning Deep Learning Large Language Models Reinforcement Learning Information Retrieval Python TensorFlow PyTorch SQL Scala Java Neural Networks Transformers A/B Testing Multi-task Learning Contextual Bandits Data Pipelines
6+ yrs exp

Senior Machine Learning Engineer, Ads Predictions

Apple Inc

Cupertino, CA 161 days ago $216,200–$324,800
Machine Learning Deep Learning Large Language Models Information Retrieval Python TensorFlow PyTorch Reinforcement Learning Contextual Bandits Multi-task Learning SQL Scala Java A/B Testing Neural Networks Transformers
6+ yrs exp

Staff Machine Learning Engineer, Ads Predictions

Apple Inc

Cupertino, CA 43 days ago $216,200–$324,800
Machine Learning Deep Learning Large Language Models Reinforcement Learning Information Retrieval Python TensorFlow PyTorch SQL Scala Java Neural Networks Transformers A/B Testing Multi-task Learning Contextual Bandits
8+ yrs exp