Skip to content

Senior Staff Machine Learning Engineer, ML Understanding

266k – 372kUnited StatesRemote10+ YOE
Summary

Leads design and implementation of advanced user modeling systems using ML and LLMs to create unified user representations for personalization across Reddit's Feeds, Search, Notifications, and Ads. Requires 10+ years in production ML, especially user understanding and recommender systems.

About the role

What you'll do

  • Design User Understanding Strategy: Define a unified user understanding framework and strategy: how users are represented (embeddings, tags, attributes, LLM-based user profile), how they are computed, stored, and exposed. Provide thought leadership in user understanding and user modeling by setting a long-term technical vision and advancing the state-of-the-art in the field.
  • Build Foundational User Models: Lead design and implementation of advanced user models, e.g. large-scale user representation learning (sequence-based, multi-interest, multi-task) that share representations across surfaces to improve personalization experience across key Reddit products e.g. Feeds, Notification, Search and Ads, balancing latency, cost, and performance.
  • Reimagine user understanding with LLM/Gen-AI: Evolve user modeling beyond traditional representations by leveraging LLMs to build richer user understanding (e.g., dynamic user profiles, intent inference, semantic reasoning over user behavior). Explore how LLMs can augment or unify embeddings, attributes, and taxonomies to enable more adaptive, interpretable, and context-aware personalization.
  • Ship Large Scale User Understanding as a System: Partner with platform teams to design and build core components for large-scale learning and serving: storage/retrieval for embeddings, feature pipelines, and APIs. Collaborate with ML/Ranking infra to ensure low-latency serving, high availability, and integration with MLOps systems.
  • Drive Cross-Team Integration & Impact: Partner with Feeds, Notification, Search and Ads teams to drive experimentation and adoption of new user understanding models with product teams across Reddit, ensuring measurable end-to-end impact on key metrics.
  • Set Technical Bar & Mentor: Mentor senior to staff engineers, lead design reviews, steward technical decisions across the user understanding domain, and champion and drive engineering processes and best practices

Who you might be

  • At least 10 years experience building and scaling production-grade ML systems, particularly in user modeling, large-scale representation learning, or recommender systems.
  • Track record of driving ambiguous, high-impact initiatives from concept to production, shaping both technical direction and execution.
  • Product- and impact-oriented: care deeply about how work moves real metrics (e.g., engagement, retention, revenue).
  • Strong fundamentals in mainstream user understanding ML approaches (e.g., representation learning, behavioral modeling, user clustering), and understand their trade-offs in real-world systems.
  • Excited about the GenAI shift and have experience (or strong intuition) applying LLMs or foundation models to evolve existing systems.
  • Think in systems, not just models: consider data, training, evaluation, serving, and adoption as a cohesive whole.
  • Influence beyond immediate team: partnering effectively with product, infra, and other ML teams.
  • Raise the technical bar: mentoring senior engineers, leading design reviews, and establishing best practices.
  • Comfortable navigating trade-offs across quality, latency, cost, and safety, especially in large-scale, user-facing systems.
Skills
Machine LearningLLMsEmbeddingsRepresentation LearningRecommender SystemsUser ModelingMLOpsLarge-scale ML SystemsFeature PipelinesPersonalization
Similar roles at this salary range
All ML Engineering jobs →
Airbnb

Senior Staff Machine Learning Engineer, Communication & Connectivity

Lead ML architecture and implementation for Airbnb's Messaging & Notifications, building recommendation engines, ranking systems, and LLM-powered experiences while mentoring engineers.

244k – 305kUnited StatesML EngineeringRemotePythonAI Systems
Traba

Staff Software Engineer

Founding Staff Applied Agent Engineer to architect and lead Traba's agentic platform, building production LLM/agent systems that integrate with customer WMS/TMS/ERP and drive industrial operations. Requires 7+ years engineering experience with 2+ years building production agent systems.

240k – 300kNew York, NY +1ML EngineeringOn-siteLLMKafka
Cribl

Staff Software Engineer, Cribl AI

Staff-level AI/ML engineer building and productionizing generative AI features across backend and frontend for Cribl's observability platform. Requires 6+ years experience, AI/ML and MLOps background, and TypeScript/JavaScript proficiency.

225k – 265kUnited StatesML EngineeringRemoteLLMsReact
Perplexity

Member of Technical Staff

ML Engineer building and optimizing production recommendation, ranking, and personalization systems that integrate LLMs for Perplexity's AI product.

220k – 405kSan Francisco, CA +1ML EngineeringOn-siteLLMsFeature Stores
Perplexity

Member of Technical Staff

Build and own multimodal AI product and platform systems across the stack at Perplexity. Requires production systems experience, full-stack capability, and strong product judgment.

220k – 405kSan Francisco, CAML EngineeringOn-siteC++Rust