Staff Machine Learning Engineer - LS Embedding

Reddit, Inc.; United States; Remote; USD 230,000 - 322,000; Full time

Job summary

An innovative company is seeking a Staff Machine Learning Engineer to lead the development of cutting-edge embedding models. This role involves guiding the technical direction for large-scale graph-based architectures and enhancing personalization across a vast ecosystem. You will collaborate with leadership to define ML roadmaps, drive innovation, and mentor a talented team. This position offers the chance to make a significant impact on AI-driven systems while working with advanced technologies such as Graph Neural Networks and deep learning techniques. If you're passionate about machine learning and eager to influence the future of personalization, this opportunity is for you.

Benefits

Medical Insurance

Dental Insurance

Vision Insurance

401(k) with Employer Match

Generous Time Off

Parental Leave

Equity in the form of Restricted Stock Units

Qualifications

• 8+ years of experience in machine learning engineering with a focus on embeddings.
• Expertise in GNNs and large-scale representation learning techniques.

Responsibilities

• Lead development of multi-entity embeddings using GNNs and transformers.
• Define ML strategy for embedding models to enhance personalization.

Skills

Machine Learning Engineering

Representation Learning

Graph Neural Networks (GNNs)

Deep Learning Techniques

Python Programming

C++ Programming

A/B Testing

Model Evaluation Frameworks

Tools

PyTorch Geometric (PyG)

Deep Graph Library (DGL)

TensorFlow

JAX

Job description

Staff Machine Learning Engineer - LS Embedding

Remote - United States

Reddit is a community of communities. It’s built on shared interests, passion, and trust and is home to the most open and authentic conversations on the internet. Every day, Reddit users submit, vote, and comment on the topics they care most about. With 100,000+ active communities and approximately 101M+ daily active unique visitors, Reddit is one of the internet’s largest sources of information. For more information, visit redditinc.com.

About the Team

The LS Embedding Team is at the forefront of building highly expressive, multi-entity embeddings that power Reddit’s recommendation systems. We go beyond standard two-tower architectures, leveraging Graph Neural Networks (GNNs), large-scale transformer models, and cutting-edge representation learning techniques to enhance personalization across Reddit’s ecosystem. Our work impacts content discovery, user engagement, and platform growth at a massive scale.

About the Role

As a Staff Machine Learning Engineer, you will own the technical direction for large-scale embedding models, guiding the development of state-of-the-art graph-based ML architectures and high-impact representation learning strategies. You will partner with leadership to define ML roadmaps, drive innovation in large-scale graph embeddings and deep learning techniques, and ensure scalable, efficient deployment of ML models in production. This role offers an opportunity to influence key AI-driven systems across Reddit while mentoring and uplifting the team’s technical capabilities.

Responsibilities

• Architect and lead the development of next-generation multi-entity embeddings, leveraging Graph Neural Networks (GNNs), transformers, and large-scale representation learning techniques.
• Define and execute the ML strategy for embedding models, identifying opportunities to enhance personalization and recommendation quality across Reddit.
• Lead research initiatives on scalable graph-based learning, self-supervised techniques, and real-time 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 embeddings 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 at the cutting edge.
• Drive technical discussions, present findings to leadership, and contribute to long-term ML planning and decision-making.

Qualifications

• 8+ years of experience in machine learning engineering, with a strong focus on representation learning, large-scale embeddings, and recommendation systems.
• Expertise in Graph Neural Networks (GNNs), graph-based representation learning, and transformer architectures.
• Deep understanding of graph theory, knowledge graphs, and complex multi-entity relationships in machine learning applications.
• Proven ability to design, implement, and optimize scalable ML architectures, from distributed training to real-time inference.
• Hands-on experience with PyTorch Geometric (PyG), Deep Graph Library (DGL), TensorFlow, JAX, and large-scale ML model optimization.
• 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.

Compensation and Benefits

Pay Transparency:

This job posting may span more than one career level.

In addition to base salary, this job is eligible to receive equity in the form of restricted stock units, and depending on the position offered, it may also be eligible to receive a commission. Additionally, Reddit offers a wide range of benefits to U.S.-based employees, including medical, dental, and vision insurance, 401(k) program with employer match, generous time off for vacation, and parental leave. To learn more, please visit redditinc.com/careers.

The base pay range for this position is:

$230,000 - $322,000 USD

Reddit is proud to be an equal opportunity employer, and is committed to building a workforce representative of the diverse communities we serve. Reddit is committed to providing reasonable accommodations for qualified individuals with disabilities and disabled veterans in our job application procedures. If you need assistance or an accommodation due to a disability, please contact us at [e-mail].

Apply for this job

* indicates a required field

First Name *

Last Name *

Email *

Phone *

Location (City) *

Resume/CV *

LinkedIn Profile

How did you hear about this job? *

Are you currently authorized to work in the U.S.? *

Do you now, or will you in the future, require immigration sponsorship to work at Reddit? *

By selecting 'I agree,' I understand that the information I have provided as part of this job application will be processed in accordance with Reddit's Candidate Privacy Policy. *

Reddit U.S. Equal Employment Information

Reddit is proud to be an equal opportunity employer, and is committed to building a workforce representative of the diverse communities we serve.