Staff Machine Learning Engineer, Retrieval

Reddit, Inc. - San Francisco, CA

Hiring: Staff Machine Learning Engineer, Retrieval Company: Reddit, Inc. Location: San Francisco, CA Job Posted Time: 2026-09-17 03:49:20 Employment Type: Remote Target Skills & Keywords : AI, Deep Learning, Machine Learning, PyTorch, TensorFlow, TestNG, Transformers About the job Experience: •7+ years of industry experience, including substantial experience building and shipping applied ML products. Required Skills: •Define the technical direction and multi-year roadmap for ads retrieval modeling in partnership with engineering, product, data science, and ads stakeholders. •Design, develop, and launch candidate-generation and retrieval models for campaigns and ads across Reddit's advertising surfaces. •Improve the retrieval stack across key modeling decisions, including objectives, labels, sampling strategies, hard-negative mining, feature design, embedding generation, candidate filtering, and retrieval depth. •Interface directly with approximate nearest-neighbor and vector retrieval systems, reasoning about recall, relevance, freshness, diversity, coverage, latency, and cost trade-offs. •Establish strong evaluation practices that connect retrieval metrics—such as recall, precision, candidate coverage, calibration, and downstream lift—to ads and user outcomes. •Lead offline analysis and online experiments, interpret ambiguous results, and translate findings into the next modeling iteration. •Partner with downstream ranking, ads platform, auction, measurement, and product teams to ensure retrieval models integrate effectively into the full ads funnel. •Write design documents, review code and model changes, and raise the quality bar for modeling, testing, observability, and production ownership. Qualifications: •7+ years of industry experience, including substantial experience building and shipping applied ML products. •Deep experience with information retrieval, candidate generation, recommender systems, ranking, or related relevance problems. •In-depth knowledge of retrieval modeling concepts, including DNN, embeddings, two-tower or dual-encoder models, approximate nearest-neighbor search, and multi-stage retrieval. •Deep experience training, evaluating, debugging, and deploying deep learning models using TensorFlow, PyTorch, or similar frameworks. •Demonstrated ownership of ML projects from problem framing and data preparation through offline evaluation, online experimentation, production launch, and iteration. •Strong command of experimental design and model evaluation, including how offline retrieval metrics relate to downstream business and user metrics. •Experience working with large-scale behavioral, contextual, or content datasets and complex feature pipelines. •Strong software engineering fundamentals and the ability to write clear, reliable, maintainable production code. •Technical leadership experience: setting direction, leading complex projects, influencing partner teams, and mentoring other engineers. •Excellent written and verbal communication, with the ability to explain complex modeling choices to technical and non-technical audiences. Compensation: •$230,000 - $322,000 / year •Flexible work environment (work from home / hybrid options) Interested candidates, please apply directly through the job posting on company's career page or try via AI auto apply on this platform. Don't miss this opportunity to join a forward-thinking team!