Skip to content
RedditReddit

Staff Machine Learning Engineer, Consumer

Leads end-to-end ML initiatives for Reddit's consumer features like recommendations, search, and AI discovery. Requires 7+ years experience with deep learning frameworks, production ML systems, and expertise in Transformers, LLMs, or recommender systems.

About the job

What You’ll Do

  • Lead end-to-end ML initiatives from ideation through production and iteration, shaping technical direction and translating product goals into scalable solutions
  • Architect, build and deploy large-scale ML systems across recommendation, search, and content/user understanding, including retrieval/ranking models, representation learnings embeddings optimizations, and LLM or GenAI-powered capabilities
  • Drive measurable impact on user engagement, discovery, and long-term value
  • Collaborate with cross-functional teams to align product and technical roadmaps and unlock key future ML capabilities
  • Stay at the forefront of AI research, evaluating and introducing new AI/ML paradigms to keep Reddit’s ML ecosystem at the cutting edge
  • Contribute to the development of best practices, guidelines, and ethical AI principles for responsible LLM development and deployment
  • Mentor and guide senior and mid-level ML engineers, fostering a culture of excellence, innovation, and knowledge sharing
  • Set technical vision and drive technical discussions, present findings to leadership, and contribute to long-term ML planning and decision-making

Required Qualifications

  • 7+ years of experience building, deploying, and operating machine learning systems in production
  • Deep understanding of machine learning methods, spanning classical approaches and modern deep learning (e.g., Transformers, GNN)
  • Expert at developing and productionizing models using TensorFlow, PyTorch, or Hugging Face Transformers
  • Experience building production-quality code incorporating testing, evaluation, and monitoring using object-oriented programming, including experience in Python and Golang
  • Experience designing and scaling ML systems, including data pipelines, feature engineering, model training/serving, and production monitoring
  • Excellent communication and collaboration skills, with the ability to discuss complex technical topics with diverse teams and translating product needs into scalable ML solutions
  • Track record of driving measurable impact through applied machine learning in real-world products

Preferred Qualifications

  • Subject matter expertise in one of the following domains:
    • Recommender systems
    • Search systems (lexical and semantic retrieval and ranking)
    • Content understanding (NLU/NLP/LLM, topic/taxonomy modeling, interest graphs or clustering, and multimodal understanding)
  • Familiarity with distributed systems and large-scale data processing frameworks (Spark, Kafka, Ray, Airflow, BigQuery, Redis)
  • Experience working with real-time systems and low-latency production environments
  • Experience with LLM/GenAI techniques, including but not limited to LLM evaluation, alignment, fine-tuning, knowledge distillation, RAG/agentic systems and productionizing LLM-powered products at scale
  • Strong experimentation rigor, with experience formulating clear hypotheses, designing actionable learning plans and building offline/online correlations
  • Advanced degree in Computer Science, Machine Learning, or related quantitative field

Skills

PyTorch, TensorFlow, Hugging Face Transformers, Python, Go, Transformers, Graph Neural Networks, Spark, Kafka, Ray, Airflow, BigQuery, Redis, LLMs, RAG

Harvey

Harvey

San Francisco, CA

Staff Software Engineer, Model Infrastructure
$231k+/yrHybrid7+ YOEML Engineering

Leads the design and operation of reliable, scalable model infrastructure powering AI inference across multiple providers. Requires 7+ years of distributed-systems engineering experience, strong programming skills, and expertise in production reliability and cloud infrastructure.

Shield AI

Shield AI

San Mateo, CA

Staff Software Engineer, Autonomy Capabilities
$234k+/yrOn-site7+ YOEML Engineering

Leads the design, implementation, integration, and field validation of tactical autonomy and multi-agent coordination capabilities for unmanned platforms. Requires 7+ years of relevant experience, production C++, technical leadership, and eligibility for a U.S. Secret clearance.

Snowflake

Snowflake

Bellevue, WA

Staff Software Engineer - Snowflake Feature Store
$236k+/yrOn-site10+ YOEML Engineering

Leads the roadmap and technical vision for Snowflake Feature Store, building reliable, high-performance machine learning platform capabilities and supporting technical execution across partner teams. Requires 10+ years of experience with data-serving infrastructure or ML platforms, plus Java and Python expertise.

Shield AI

Shield AI

Washington, DC
Staff Engineer, Autonomy Capabilities – Maritime
$221k+/yrOn-site7+ YOEML Engineering

Leads development and integration of advanced maritime autonomy for USVs, UUVs, and cooperating UAVs, including motion planning, localization, safety, and multi-agent coordination. Requires staff-level technical leadership, substantial robotics experience, C++ and Python proficiency, and eligibility for a SECRET clearance.

Ironclad

Ironclad

San Francisco, CA

Senior Staff Software Engineer, Agentic Search
$220k+/yrHybrid10+ YOEML Engineering

Leads architecture and technical direction for agentic search systems combining LLMs, retrieval, and content-understanding pipelines for contract intelligence. The role requires 10+ years building production systems, deep search or LLM expertise, and strong cross-team technical leadership.