Staff Machine Learning Engineer, Ads ML Efficiency

Details

Reddit’s 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. The team focuses on improving developer productivity, reducing infrastructure costs, increasing hardware utilization, and accelerating experimentation across Reddit’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 Reddit ML engineers.
• Drive technical strategy for ML platform scalability, reliability, and cost efficiency.

Requirements

• 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, such as 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 Qualifications

• Experience with large-scale recommendation, ranking, generative AI, or foundation model systems.
• Experience with distributed training frameworks such as PyTorch Distributed, Ray, TensorFlow, or 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 building real-time ML inference applications.

What Success Looks Like

• ML engineers can move from idea to experiment faster.
• Training and inference costs decrease and 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 supporting workspace, professional development, and caregiving.
• Family planning support.
• Gender-affirming care.
• Mental health and coaching benefits.
• Group personal pension scheme with employer match.
• Private medical and dental scheme.
• Income replacement programs.
• Bike-to-work scheme.
• Flexible vacation and paid volunteer time off.
• Generous paid parental leave.
• Equity in the form of restricted stock units.
• U.S.-based employees may receive medical, dental, and vision insurance, a 401(k) program with employer match, vacation, and parental leave.

Compensation

The base salary range for this position is $230,000–$322,000 USD. The role may also be eligible for equity and, depending on the position offered, commission.