Staff Machine Learning Engineer, ML Efficiency

This is a fully remote position, open to applicants in United Kingdom.

📋 Description

• Design and develop systems that enhance the efficiency of ML training and inference tasks.

• Create tools that assist ML engineers in debugging, profiling, optimizing, and monitoring model performance.

• Enhance GPU and overall resource utilization through effective scheduling, resource management, caching, and workload optimization.

• Collaborate with ML researchers and product teams to pinpoint bottlenecks and implement performance enhancements.

• Construct benchmarking frameworks and performance dashboards for both training and serving systems.

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

• Lead cross-functional projects aimed at increasing the productivity of Reddit's ML engineers.

• Shape the technical strategy for the scalability, reliability, and cost-effectiveness of the ML platform.

⛳️ Requirements

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

• Over 5 years of experience in software engineering.

• Strong expertise in Python.

• Proficiency in at least one systems programming language (Go, C++, Rust, or Java) is preferred.

• Experience in building distributed systems on a large scale.

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

• Comprehensive understanding of performance engineering and systems optimization.

• Excellent debugging and profiling abilities.

🏝️ Benefits

• Global benefit programs tailored to your lifestyle, encompassing workspace, professional development, and caregiving support.

• Family planning assistance.

• Gender-affirming care.

• Mental health and coaching benefits.

• Group personal pension scheme with employer contribution.

• Private medical and dental coverage.

• Income replacement programs.

• Bike-to-work scheme.

• Flexible vacation and paid volunteer time off.

• Generous paid parental leave.