Staff Machine Learning Engineer, Ads ML Efficiency

New

RedditML infrastructure

You can work remotely from anywhere in the US or Canada.Full-TimeStaff

Salary$230,000 — $322,000 USD

Apply NowOpens the employer's application page

Job Details

Hold a BS, MS, or PhD in Computer Science or a related field.
Have 5+ years of software engineering experience.
Demonstrate strong proficiency in Python.
Have proficiency in at least one systems language; Go, C++, Rust, or Java proficiency is preferred.
Have experience building distributed systems at scale.
Have experience with machine learning infrastructure, training systems, or model serving platforms.
Bring a deep understanding of performance engineering and systems optimization.
Have strong debugging and profiling skills.
Preferred: Experience with large-scale recommendation, ranking, generative AI, or foundation model systems.
Preferred: Experience with distributed training frameworks such as PyTorch Distributed, Ray, TensorFlow, or Spark.
Preferred: Familiarity with GPU architectures and performance analysis tools.
Preferred: Experience optimizing cloud infrastructure costs across large ML workloads.

Design and build systems that improve the efficiency of ML training and inference workloads.
Develop tooling for ML engineers to 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.

View Full Description & ApplyYou'll be redirected to the employer's site