Skip to content
RedditRedditUnited States

Staff Machine Learning Systems Engineer, Embeddings Platform

Staff ML Systems Engineer leading large-scale embedding and recommendation model architecture, distributed training, and real-time serving. Owns ML strategy and mentors engineers on personalization systems.

253k – 355k
Remote8+ YOEML Engineering

About the role

What You’ll Do

  • Architect and lead the development of next-generation, large-scale machine learning techniques.
  • Define and execute the ML strategy, identifying opportunities to enhance personalization and recommendation quality across Reddit.
  • Lead research initiatives on scalable machine learning systems and real-time model adaptation, bringing cutting-edge advancements into production.
  • Partner with ML infrastructure teams to build high-performance, distributed training systems that efficiently scale across multiple GPUs and cloud environments.
  • Establish and optimize real-time serving architectures for large-scale embeddings, ensuring low-latency inference and high throughput.
  • Collaborate cross-functionally with teams in Feed Ranking, Ads, Content Understanding, and Core ML to integrate ML models into Reddit’s key AI-driven systems.
  • Mentor and guide senior and mid-level ML engineers, fostering a culture of excellence, innovation, and knowledge sharing.
  • Stay at the forefront of AI research, evaluating and introducing new modeling paradigms to keep Reddit’s ML ecosystem cutting-edge.
  • Drive technical discussions, present findings to leadership, and contribute to long-term ML planning and decision-making.

Who You Might Be

  • 8+ years of experience in machine learning engineering, with a strong focus on large-scale ML systems and recommendation or personalization systems.
  • Expertise in modern deep learning architectures, including sequence models and foundational models.
  • Deep understanding of complex multi-entity relationships in machine learning applications and how they are modeled in large-scale systems.
  • Proven ability to design, implement, and optimize scalable ML architectures, from distributed training to real-time inference.
  • Strong software engineering skills in Python, C++, or similar languages, with experience in ML infrastructure, high-performance computing, and cloud-based ML pipelines.
  • Demonstrated leadership in driving ML strategy, mentoring engineers, and influencing cross-functional teams.
  • Experience with A/B testing, model evaluation frameworks, and real-time feedback loops in large-scale production systems.
  • Excellent communication skills, with the ability to effectively present complex ML concepts to technical and non-technical stakeholders.

Benefits

  • Comprehensive Healthcare Benefits and Income Replacement Programs
  • 401k with Employer Match
  • Global Benefit programs that fit your lifestyle, from workspace to professional development to caregiving support
  • Family Planning Support
  • Gender-Affirming Care
  • Mental Health & Coaching Benefits
  • Flexible Vacation & Paid Volunteer Time Off
  • Generous Paid Parental Leave

Skills

PythonC++Deep LearningDistributed TrainingReal-Time InferenceA/B TestingModel EvaluationML InfrastructureHigh-Performance ComputingCloud-Based Ml Pipelines

Similar roles

ML Engineering jobs
Harper

Staff Engineer, Harness Engineering

HarperSan Francisco, CA

As a Staff Engineer, Harness Engineering, you will own the meta-harness that powers all Harper AI agents, both product and coding. This involves designing and building the agent loop, execution environment, tool layer, and model-provider abstraction to ensure efficient and reliable agent operation.

253k – 308k
On-site8+ YOEML Engineering
Coinbase

Senior Staff Software Engineer, Legal Automation

CoinbaseUnited States

Senior Staff Software Engineer building an AI agent platform and automated workflows to transform Coinbase's Legal organization. Architect production-grade LLM and multi-agent systems that replace manual legal processes such as agreement redlining and governance.

254k – 299k
Remote12+ YOEML Engineering
Postman

Member of Technical Staff, AI Agent Development Lead

PostmanSan Francisco, CA +3

Leads design, development, and deployment of AI agents using language models and ML frameworks. Drives scalable, safe AI systems while mentoring teams and collaborating cross-functionally. Requires strong Python skills and AI leadership experience.

256k – 276k
HybridML Engineering
Axion

Staff Software Engineer, Agentic Platform

AxionSan Francisco, CA +1

Senior individual contributor architecting and scaling agentic LLM systems that turn messy manufacturing data into reliable root-cause insights. Owns orchestration, retrieval, evaluation, and guardrails for non-deterministic production systems.

250k – 270k
Hybrid7+ YOEML Engineering
The Voleon Group

Member of Research Staff, Optimization

The Voleon GroupBerkeley, CA +1

Conduct optimization research and implement large-scale constrained optimization models that drive real-time trading decisions, working across the full research lifecycle from theory to production. Requires PhD-level coursework and strong applied research background in optimization.

250k – 275k
Hybrid7+ YOEML Engineering