Staff Machine Learning Infrastructure Engineer, Embedding Platform
RedditSocial Media company
United States$253,300 - $354,600 USDLead
Fidelity Investments
Andreessen Horowitz
Sequoia Capital
Coatue
Vy Capital
Sam Altman
Data & AI
About the role
TL;DR
Lead the development of large-scale ML infrastructure for Reddit's embedding platform.
- •The LS Embedding Machine Learning Platform team is at the forefront of building highly expressive machine learning models that power Reddit’s recommendation systems.
- •We go beyond standard retrieval and ranking architectures, leveraging modern deep learning approaches and scalable model designs to enhance personalization across Reddit’s ecosystem.
- •Key Responsibilities 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.
- •Partner with ML infrastructure teams to build high-performance, distributed training systems.
- •Establish and optimize real-time serving architectures for large-scale embeddings.
- •Requirements 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.
- •Proven ability to design, implement, and optimize scalable ML architectures.
- •Strong software engineering skills in Python, C++, or similar languages.
Required skills
PythonC++TensorFlowPyTorchscikit-learnKubernetesAWSGoogle CloudAzureDockerCI/CDA/B TestingNLPLLMs
Domain expertise
ai
Benefits & perks
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
Tech stack
PythonC++TensorFlowPyTorchscikit-learnKubernetesAWSGoogle CloudAzureDockerCI/CDA/B TestingNLPLLMs