Skip to content

Staff Machine Learning Engineer, Consumer

230k – 322kUnited StatesRemote7+ YOE
Summary

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 role

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
PyTorchTensorFlowHugging Face TransformersPythonGolangTransformersGraph Neural NetworksSparkKafkaRayAirflowBigQueryRedisLLMRAG
Similar roles at this salary range
All ML Engineering jobs →
Databricks

Staff Software Engineer, AI Runtime

Staff Software Engineer building and scaling Databricks' managed large-scale GPU training platform (AIR). Focus on distributed training performance, scheduling, fault tolerance, and developer experience for thousands of accelerators.

190k – 265kMountain View, CA +1ML EngineeringOn-siteFSDPRoCE
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
Traba

Senior Software Engineer

Founding Senior Applied Agent Engineer building production LLM agent systems that automate supply chain workflows. Requires 5+ years engineering experience with 1+ year shipping LLM/agent features, strong Python/TypeScript skills, and hands-on agent stack experience.

200k – 240kNew York, NY +1ML EngineeringOn-sitePythonNode.js
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