Staff Machine Learning Engineer, Ads ML Efficiency

United States Digital Space LLC; United States; Remote; USD 230,000 - 322,000; Full time

Benefits offered by this job

Global Benefits

Medical, Dental, Vision

401(k) matching

Paid Parental Leave

Flexible vacation

Job summary

Reddit is seeking a senior ML Platform Engineer to design and optimize scalable training and inference systems across our global infrastructure. You will build tooling, improve scheduling, and partner with researchers to accelerate experiments.

You will lead distributed systems work, drive performance engineering, and help reduce costs while increasing GPU utilization. This role offers fully remote work in the US/Canada and a comprehensive benefits package.

Qualifications

• BS, MS, or PhD in Computer Science or a related field.
• 5+ years of software engineering experience.
• Strong proficiency in Python.
• Proficiency in at least one systems language (Go, C++, Rust, or Java).
• Experience building distributed systems at scale.
• Experience with machine learning infrastructure, training systems, or model serving platforms.
• Deep understanding of performance engineering and systems optimization.
• Strong debugging and profiling skills.

Responsibilities

• Design and build systems that improve the efficiency of ML training and inference workloads.
• Develop tooling that helps ML engineers debug, profile, optimize, and monitor model performance.
• Improve GPU and general resource utilization through scheduling, resource management, caching, and workload optimization.
• Partner with ML researchers and product teams to identify bottlenecks and drive performance improvements.
• Build benchmarking frameworks and performance dashboards for training and serving systems.
• Optimize distributed training infrastructure, data pipelines, and model serving architectures.
• Lead cross-functional initiatives that improve the productivity of the company ML engineers.
• Drive technical strategy for ML platform scalability, reliability, and cost efficiency.

Skills

Python

Go

C++

Rust

Java

Distributed systems

Performance engineering

Debugging

Profiling

Education

BS/MS/PhD in Computer Science

Tools

PyTorch

Ray

TensorFlow

Spark

Job description

the company 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, the company 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, the company is one of the internet’s largest sources of information. For more information, visit www.redditinc.com.

Location:

the company has a flexible first workforce. Don't live near our office? No worries: you can work remotely from anywhere in the US or Canada.

About the Team

The ML Efficiency team builds the infrastructure, tooling, and optimization systems that enable machine learning engineers and researchers to train, evaluate, deploy, and operate models efficiently at scale. We focus on improving developer productivity, reducing infrastructure costs, increasing hardware utilization, and accelerating experimentation across the company’s ML ecosystem.

Responsibilities

• Design and build systems that improve the efficiency of ML training and inference workloads.

• Develop tooling that helps ML engineers debug, profile, optimize, and monitor model performance.

• Improve GPU and general resource utilization through scheduling, resource management, caching, and workload optimization.

• Partner with ML researchers and product teams to identify bottlenecks and drive performance improvements.

• Build benchmarking frameworks and performance dashboards for training and serving systems.

• Optimize distributed training infrastructure, data pipelines, and model serving architectures.

• Lead cross-functional initiatives that improve the productivity of the company ML engineers.

• Drive technical strategy for ML platform scalability, reliability, and cost efficiency.

QualificationsRequired

• BS, MS, or PhD in Computer Science or a related field.

• 5+ years of software engineering experience.

• Strong proficiency in Python

• Profiency in at least one systems language (Go, C++, Rust, or Java) preferred

• Experience building distributed systems at scale.

• Experience with machine learning infrastructure, training systems, or model serving platforms.

• Deep understanding of performance engineering and systems optimization.

• Strong debugging and profiling skills.

Preferred

• Experience with large-scale recommendation, ranking, generative AI, or foundation model systems.

• Experience with distributed training frameworks such as PyTorch Distributed, Ray, Tensorflow, Spark

• Familiarity with GPU architectures and performance analysis tools.

• Experience optimizing cloud infrastructure costs across large ML workloads.

• Contributions to internal platforms used by multiple ML teams.

• Experience with building real time ML inference applications

What Success Looks Like

• ML engineers can move from idea to experiment faster.

• Training and inference costs decrease, performance increases, while model quality is maintained or improved.

• GPU utilization and cluster efficiency increase.

• Platform reliability improves as ML workloads scale.

• Teams spend less time managing infrastructure and more time building models.

• Average recommendation model size increases.

Benefits:

• 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

• Group Personal Pension Scheme with Employer match

• Private Medical and Dental Scheme

• Income Replacement Programs

• Bike to Work scheme

• Flexible Vacation & Paid Volunteer Time Off

• Generous Paid Parental Leave

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, the company 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:

$230,000—$322,000 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.

*the company is proud to be an equal opportunity employer, and is committed to building a workforce representative of the diverse communities we serve. the company 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.*