Senior Machine Learning Infrastructure Engineer, Embedding Platform
RedditSocial Media company
Remote$190,800 - $267,100 USDSenior
Fidelity Investments
Andreessen Horowitz
Sequoia Capital
Coatue
Vy Capital
Sam Altman
Data & AI
About the role
TL;DR
Build large-scale learning systems for recommendation and personalization on Reddit.
- •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 Design, train, and improve large-scale machine learning platforms for recommendation or personalization systems.
- •Own and deliver major ML systems components end to end, from problem framing through production rollout.
- •Build and optimize end-to-end ML pipelines spanning data preparation, feature generation, training, evaluation, and deployment.
- •Improve distributed training, model efficiency, and online inference performance.
- •Apply modern modeling approaches including sequence modeling and related foundation-model techniques to Reddit use cases.
- •Requirements 5+ years of experience in machine learning engineering, with a strong focus on large-scale ML infrastructure and recommendation or personalization systems.
- •Expertise in modern deep learning architectures, including sequence models and foundational models.
- •Experience building or scaling ML platform for large datasets and high-traffic production environments.
- •Demonstrated ability to independently scope and execute ambiguous technical work, while owning high-quality implementation details.
- •Solid understanding of distributed training and inference concepts, such as data parallelism, model parallelism, pipeline parallelism, or related optimization techniques.
Required skills
PythonPyTorchTensorFlowSQLLLMsMLflowKubeflowSageMakerVertex AI
Nice-to-have skills
scikit-learnKerasOpenAI APIAnthropic APIGemini APILangChainHugging FaceComputer VisionNLPONNX
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
PythonPyTorchTensorFlowSQLGitLinux