Reddit

Senior Machine Learning Infrastructure Engineer, Embedding Platform

Reddit
Apply
27 days ago
Remote, United StatesSenior
H1B sponsor

Base Salary

$191k - $267k/yr

Responsibilities

  • Design, train, and improve large-scale machine learning platforms for recommendation and personalization systems.
  • Own major ML system components end to end, from problem framing through production rollout.
  • Build and optimize pipelines for data preparation, feature generation, training, evaluation, and deployment.
  • Improve distributed training, model efficiency, and online inference performance.
  • Apply sequence modeling and related foundation-model techniques to Reddit use cases.
  • Develop reliable serving and monitoring patterns for low-latency, high-throughput production ML systems.
  • Collaborate with product, relevance, ads, ML platform, and core ML teams.
  • Drive offline and online evaluation, experimentation, model diagnostics, and feedback-loop improvements.
  • Contribute through code, design reviews, documentation, and operational excellence.

Requirements

  • At least 5 years of machine learning engineering experience focused 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 platforms for large datasets and high-traffic production environments.
  • Ability to independently scope and execute ambiguous technical work while owning high-quality implementation details.
  • Understanding of distributed training and inference concepts, including data parallelism, model parallelism, and pipeline parallelism.
  • Proficiency in Python and experience with modern ML frameworks such as PyTorch or TensorFlow.
  • Strong software engineering fundamentals in system design, debugging, testing, and performance optimization.
  • Experience with A/B testing, model evaluation frameworks, and real-time feedback loops in large-scale production systems.
  • Excellent communication skills for presenting complex ML concepts to technical and non-technical stakeholders.

Benefits

  • Comprehensive healthcare benefits and income replacement programs.
  • 401(k) with employer match.
  • Global benefits supporting workspace, professional development, and caregiving.
  • Family planning support.
  • Gender-affirming care.
  • Mental health and coaching benefits.
  • Flexible vacation and paid volunteer time off.
  • Generous paid parental leave.
  • Remote role, as indicated by the #LI-Remote designation.

Tech Stack

PythonPyTorchTensorFlow
Reddit

About Reddit

5,001-10,000 employees

Reddit is the heart of the internet, where millions of people get together to talk about any topic imaginable. Share, vote, and decide what matters in everything from breaking news, to fandoms, lifehacks, gaming, sports, health, and the internet’s cutest animals. With over 100,000 subreddit communities about every topic you could think of (and thousands more you couldn’t), whatever it is, there’s a place for you on Reddit. Interested in joining our growing team? Check out redditinc.com/careers.

Contact me