Staff Machine Learning Engineer - LS Embedding

Job Description

Reddit is seeking a Staff Machine Learning Engineer to join its LS Embedding Team. This team focuses on building multi-entity embeddings to enhance Reddit's recommendation systems. The engineer will be responsible for the technical direction of large-scale embedding models, guiding the development of graph-based ML architectures and representation learning strategies. They will partner with leadership to define ML roadmaps and drive innovation in graph embeddings and deep learning techniques. This role involves influencing AI-driven systems across Reddit and mentoring other engineers.Responsibilities:

• Architect and lead the development of next-generation multi-entity embeddings.
• Define and execute the ML strategy for embedding models.
• Lead research initiatives on scalable graph-based learning and self-supervised techniques.
• Partner with ML infrastructure teams to build high-performance, distributed training systems.
• Establish and optimize real-time serving architectures for large-scale embeddings.
• Collaborate cross-functionally with teams in Feed Ranking, Ads, Content Understanding, and Core ML.
• Mentor and guide senior and mid-level ML engineers.
• Stay at the forefront of AI research.
• Drive technical discussions and contribute to long-term ML planning.

Requirements:

• 8+ years of experience in machine learning engineering.
• Expertise in Graph Neural Networks (GNNs) and transformer architectures.
• Deep understanding of graph theory and knowledge graphs.
• Proven ability to design and optimize scalable ML architectures.
• Hands-on experience with PyTorch Geometric (PyG), Deep Graph Library (DGL), TensorFlow, JAX.
• Strong software engineering skills in Python, C++, or similar languages.
• Demonstrated leadership in driving ML strategy and mentoring engineers.
• Experience with A/B testing and model evaluation frameworks.
• Excellent communication skills.

Reddit offers:

• Comprehensive Healthcare Benefits and Income Replacement Programs
• 401k Match
• Family Planning Support
• Gender-Affirming Care
• Mental Health & Coaching Benefits
• Flexible Vacation & Reddit Global Days off
• Generous paid Parental Leave
• Paid Volunteer time off