[Remote] Senior Machine Learning Engineer, Ads ML Efficiency

Note: The job is a remote job and is open to candidates in USA. Reddit is hiring a Senior Machine Learning Engineer for its Ads ML Efficiency team. The role optimizes the performance, scalability, and cost efficiency of production machine learning workloads, partnering with model and platform teams to identify bottlenecks and deliver measurable improvements.

Responsibilities

• Own end-to-end performance optimization initiatives for high-impact production ML workloads, from bottleneck identification and implementation through validation and production rollout
• Profile and diagnose ML performance bottlenecks across model execution, frameworks, runtimes, memory, CPU/GPU utilization, and infrastructure using rigorous benchmarking and measurement
• Design and implement performance improvements involving inference execution, runtime behavior, batching, memory management, precision, compiler integration, custom operators, kernels, or other relevant techniques
• Improve ML training and serving efficiency and scalability, optimizing latency, throughput, hardware utilization, and cost while maintaining model quality and production reliability
• Support faster and safer model launches through performance benchmarking, load testing, regression detection, and launch-readiness improvements
• Partner with model engineers and platform teams to identify high-leverage opportunities, integrate optimizations safely, and validate measurable end-to-end impact
• Build reusable performance capabilities, including benchmarks, tooling, optimization patterns, and engineering guidance that accelerate future model development and deployment
• Contribute to technical direction and mentor engineers through hands-on collaboration, code reviews, performance debugging, and knowledge sharing

Skills

• Hands-on ML performance engineering experience: Demonstrated ability to diagnose and implement meaningful performance improvements in production or production-relevant ML workloads
• Deep expertise in at least one ML performance area, such as inference execution, ML runtimes, compiler integration, GPU/kernel optimization, memory systems, or training execution
• Strong profiling and benchmarking skills: Ability to identify bottlenecks systematically, establish reliable baselines, and validate measurable end-to-end improvements
• Strong software engineering fundamentals: Experience implementing, debugging, testing, and maintaining performance-sensitive systems, with proficiency in Python and relevant systems programming technologies
• ML systems understanding: Familiarity with production ML frameworks and the performance trade-offs across model execution, runtime, memory, hardware, and infrastructure
• Production engineering judgment: Ability to balance performance improvements with numerical correctness, model quality, reliability, maintainability, and safe rollout
• Good customer and platform instincts: can solve concrete bottlenecks while keeping maintainability, adoption, and future reuse in mind
• Strong communication: able to explain tradeoffs clearly to engineers and partner teams
• Experience optimizing large-scale inference or training workloads, particularly in recommendation systems, ranking, search, or retrieval
• Experience with PyTorch, including performance optimization using technologies such as Dynamo, Inductor, AOTI, torch.export, or custom operators
• Experience with inference runtimes, GPU execution, or performance technologies such as TensorRT, ONNX Runtime, CUDA, Triton, LLVM/MLIR, or equivalent systems
• Experience with hardware-aware optimization, including batching, memory management, operator fusion, mixed precision, quantization, or accelerator efficiency
• Experience with distributed-training performance, including GPU utilization, communication, data loading, memory efficiency, or training step-time optimization
• Experience building performance benchmarks, profiling tools, regression detection, launch-readiness capabilities, or cost observability toolings for ML workloads
• Experience working across ML modeling, serving, and platform infrastructure teams
• Specialized compiler internals or custom CUDA/Triton kernel development

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
• Eligible to receive equity in the form of restricted stock units
• Depending on the position offered, may also be eligible to receive a commission
• U.S.-based employees: medical, dental, and vision insurance
• U.S.-based employees: 401(k) program with employer match
• U.S.-based employees: generous time off for vacation
• U.S.-based employees: parental leave

Company Overview

Company H1B Sponsorship