Role overview

Staff Machine Learning Engineer, ML Efficiency

Requirements and responsibilities

Readable role content extracted into sections for faster review.

Details

  • 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.
  • 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.
  • 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
  • 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.
  • 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
  • Private Pension plan with Employer-matching
  • 100% employer-sponsored group medical plan
  • Income Replacement Programs
  • Flexible Vacation & Paid Volunteer Time Off
  • Generous Paid Parental Leave
Similar roles

Keep a backup shortlist.

Browse stack
FocusAds EngineeringRole area
Seniority signalLeadCandidate level
StackJava, Python, SparkPrimary skills
Location2 accepted countriesEligibility

Stack

Use these tags to compare similar remote roles.

Location eligibility

Candidates should apply only when their profile country is listed here.

Your profileCountry not setSign in to check your country against this role.

Hiring flow

WithMira shows the role, then sends candidates to the company application.

1Check role fit, stack, and location eligibility in WithMira.
2Open the company application page from the tracked apply link.
3Save the role or subscribe for similar opportunities before leaving.
Apply on company siteCompany siteOpen link