Remote Jobs RockRemote Jobs Rock

Senior Machine Learning Infrastructure Engineer, Embedding Platform

๐Ÿ•’ 24 days ago
Machine Learning EngineeringLarge-scale ML InfrastructureRecommendation SystemsPersonalization Systems

๐Ÿ“œ Description

  • 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.
  • Develop reliable serving and monitoring patterns for low-latency, high-throughput production ML systems.
  • Work with cross-functional partners to deliver measurable improvements in user experience and business impact.

๐Ÿ› ๏ธ 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.
  • Proficiency in Python and experience with modern ML frameworks such as PyTorch, TensorFlow, or similar.
  • Strong software engineering fundamentals, including 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, with the ability to effectively present complex ML concepts to technical and non-technical stakeholders.

โœจ Benefits

  • 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
  • Mental Health & Coaching Benefits
  • Flexible Vacation & Paid Volunteer Time Off
  • Generous Paid Parental Leave
Full job description
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 130 million daily active unique visitors, Reddit is one of the internetโ€™s largest sources of information. For more information, visit www.redditinc.com.

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. Our work impacts content discovery, user engagement, and platform growth at a massive scale.

About the Role

As a Senior Machine Learning Infrastructure Engineer, you will work across both model development and ML platform to build large-scale learning systems that improve recommendation and personalization on Reddit. At the senior level, you will own major technical components end to end: designing models, implementing training and evaluation pipelines, and driving production deployment in close partnership with ML platform, product, and cross-functional ML teams.

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.
  • Develop reliable serving and monitoring patterns for low-latency, high-throughput production ML systems.
  • Work with cross-functional partners across product, relevance, ads, and core ML teams to deliver measurable improvements in user experience and business impact.
  • Drive rigorous offline and online evaluation, including experimentation, model diagnostics, and feedback-loop improvement.
  • Contribute to engineering quality through strong code, design reviews, documentation, and operational excellence.

Qualifications

  • 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.
  • Proficiency in Python and experience with modern ML frameworks such as PyTorch, TensorFlow, or similar.
  • Strong software engineering fundamentals, including 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, with the ability to effectively present complex ML concepts to technical and non-technical stakeholders.

Benefits:

  • 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

#LI-Remote

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 https://www.redditinc.com/careers/.

To provide greater transparency to candidates, we share base salary ranges for all US-based job postings regardless of state. We set standard base pay ranges for all roles based on function, level, and country location, benchmarked against similar stage growth companies. Final offer amounts are determined by multiple factors including, skills, depth of work experience and relevant licenses/credentials, and may vary from the amounts listed below.

The base salary range for this position is:
$190,800-$267,100 USD

In select roles and locations, the interviews will be recorded, transcribed and summarized by artificial intelligence (AI). You will have the opportunity to opt out of recording, transcription and summarization prior to any scheduled interviews.

During the interview, we will collect the following categories of personal information: Identifiers, Professional and Employment-Related Information, Sensory Information (audio/video recording), and any other categories of personal information you choose to share with us. We will use this information to evaluate your application for employment or an independent contractor role, as applicable. We will not sell your personal information or disclose it to any third party for their marketing purposes. We will delete any recording of your interview promptly after making a hiring decision. For more information about how we will handle your personal information, including our retention of it, please refer to our Candidate Privacy Policy for Potential Employees and Contractors.

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, due to a disability, you need an accommodation during the interview process, please let your recruiter know.

ZoomInfo

Senior Machine Learning Engineer

ZoomInfo๐Ÿ‘ฅ 10,000+ employees๐Ÿข Software Development๐Ÿค B2B
๐Ÿ•’ yesterday

As a Senior Machine Learning Engineer, you will design and implement large-scale recommendation systems, optimize NLP models, and manage MLOps infrastructure to enhance AI-driven solutions.

Machine LearningNatural Language ProcessingRecommendation SystemsMlops
Databricks

Sr. Forward Deployed Engineer (FDE) - Retail

Databricks๐Ÿ‘ฅ 10,000+ employees๐Ÿข Software Development๐Ÿค B2B
๐Ÿ•’ yesterday

As a Forward Deployed Engineer (FDE), you will lead the design and implementation of data and AI solutions for customers using the Databricks platform, ensuring impactful project delivery.

Data EngineeringData PlatformsAnalyticsSoftware Engineering
Zscaler

Staff Machine Learning Engineer - Data Lake, Anomaly Detection

Zscaler๐Ÿ‘ฅ 10,000+ employees๐Ÿข Computer And Network Security๐Ÿค B2B
๐Ÿ•’ yesterday

As a Staff Machine Learning Engineer, you will enhance Zscaler's cloud security platform by integrating advanced AI technologies and designing scalable production systems.

AI/ML TechnologiesGenaiCloud PlatformsData Lakes
CoreWeave

Staff Software Engineer, Infrastructure - Marimo

CoreWeave๐Ÿ‘ฅ 10,000+ employees๐Ÿข IT Services And IT Consulting๐Ÿค B2B
๐Ÿ”ฅ 10 hours ago

As a Staff Engineer on Marimo's molab team, you will co-design and implement the backend architecture of molab, focusing on high availability, low latency, and system stability.

Software EngineeringBackend ArchitectureHigh Availability SystemsLow Latency Communication

Trusted by Remote Workers