# Staff Machine Learning Infrastructure Engineer, Embedding Platform

**Company:** [Reddit](https://hotfix.jobs/companies/reddit)
**Location:** Remote
**Role:** ML Engineering
**Salary:** $253k – $355k/yr
**Experience:** 8+ years
**Skills:** Python, C++, Deep Learning, sequence models, Foundation Models, Distributed Training, gpu computing, real-time inference, machine learning infrastructure, cloud ml pipelines, Recommendation Systems, A/B Testing, Model Evaluation, Embeddings, high-performance computing
**Posted:** 2026-08-12

> Leads the technical direction of large-scale ML infrastructure for embedding, recommendation, and personalization systems. The role requires 8+ years of ML engineering experience, expertise in deep learning and distributed training, and strong leadership across research, infrastructure, and production deployment.

## Job Description

## Responsibilities
- Architect and lead the development of next-generation, large-scale machine learning techniques.
- Define and execute ML strategy to improve personalization and recommendation quality.
- Lead research initiatives on scalable machine learning systems and real-time model adaptation, bringing advancements into production.
- Partner with ML infrastructure teams to build high-performance, distributed training systems that 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 with Feed Ranking, Ads, Content Understanding, and Core ML teams to integrate models into AI-driven systems.
- Mentor and guide ML engineers while fostering technical excellence, innovation, and knowledge sharing.
- Evaluate and introduce new modeling paradigms and contribute to long-term ML planning.
- Drive technical discussions and present findings to leadership and stakeholders.

## Requirements
- **8+ years** of experience in machine learning engineering, focused 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 and their modeling in large-scale systems.
- Experience designing, implementing, and optimizing scalable ML architectures, distributed training, and real-time inference.
- Strong software engineering skills in Python, C++, or similar languages.
- Experience with 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 for presenting complex ML concepts to technical and non-technical stakeholders.

## Compensation and Benefits
- Base salary: **$253,300–$354,600 USD**
- Equity in the form of restricted stock units; select positions may also be eligible for commission.
- 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.
- Paid parental leave.

## Similar roles

- [Senior Staff Software Engineer, Finance Automation](https://hotfix.jobs/jobs/0e8bc184-b379-45d7-bdfa-734fdf0de06d) - Coinbase - Remote - $254k – $299k/yr
- [Senior Staff Software Engineer, Legal Automation](https://hotfix.jobs/jobs/7870d3e1-ebec-4f7b-9ae5-c62926a06d6e) - Coinbase - Remote - $254k – $299k/yr
- [Staff Frontier Agents Engineer](https://hotfix.jobs/jobs/5fdd565f-7277-4721-afe0-bb496b56b6d2) - Scale AI - San Francisco, CA - $252k – $315k/yr
- [Member of Technical Staff, AI Agent Development Lead](https://hotfix.jobs/jobs/06642e42-50eb-454d-8ec8-f1ab0104efca) - Postman - San Francisco, CA - $256k – $276k/yr
- [Staff ML Engineer, Agent Training & Environments](https://hotfix.jobs/jobs/faee3b40-f74e-43a6-97d5-b2eb7571cc64) - Labelbox - San Francisco, CA - $250k – $280k/yr

**Apply:** https://hotfix.jobs/jobs/9b4387bf-ac16-48ba-810b-2e6a612e3617
**Canonical:** https://hotfix.jobs/jobs/9b4387bf-ac16-48ba-810b-2e6a612e3617