Staff Machine Learning Infrastructure Engineer, Embedding Platform

Details

The LS Embedding Machine Learning Platform team builds highly expressive machine learning models that power Reddit’s recommendation systems. The team leverages modern deep learning approaches and scalable model designs to improve personalization, content discovery, user engagement, and platform growth.

As a Staff Machine Learning Infrastructure Engineer, you will own the technical direction for large-scale machine learning platforms, guide the development of advanced deep learning architectures and high-impact ML systems, define ML roadmaps, drive innovation in scalable model design and training, and ensure the efficient and reliable deployment of ML models in production. You will also mentor engineers and influence AI-driven systems across Reddit.

Responsibilities

• Architect and lead the development of next-generation, large-scale machine learning techniques.
• Define and execute ML strategy, identifying opportunities to improve personalization and recommendation quality.
• Lead research initiatives involving scalable machine learning systems and real-time model adaptation, bringing advancements into production.
• Partner with ML infrastructure teams to build high-performance distributed training systems that 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 with Feed Ranking, Ads, Content Understanding, and Core ML teams to integrate models into Reddit’s AI-driven systems.
• Mentor and guide senior and mid-level ML engineers.
• Evaluate and introduce new modeling paradigms and advances in AI research.
• Drive technical discussions, present findings to leadership, and contribute to long-term ML planning and decision-making.

Requirements

• 8+ years of experience in machine learning engineering, with a 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 their modeling in large-scale systems.
• Experience designing, implementing, and optimizing scalable ML architectures, including distributed training and real-time inference.
• Strong software engineering skills in Python, C++, or similar languages.
• Experience with 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, including the ability to present complex ML concepts to technical and non-technical stakeholders.

Benefits

• Comprehensive healthcare benefits and income replacement programs.
• 401(k) with employer match.
• Global benefit programs supporting workspace, professional development, caregiving, and lifestyle needs.
• Family planning support.
• Gender-affirming care.
• Mental health and coaching benefits.
• Flexible vacation and paid volunteer time off.
• Generous paid parental leave.
• Equity in the form of restricted stock units.

The base salary range for this position is $253,300–$354,600 USD. The role may span more than one career level and may also be eligible for commission depending on the position offered. Reddit-based employees may receive medical, dental, and vision insurance, a 401(k) program with employer match, vacation time, and parental leave.