Senior Machine Learning Engineer, Ads ML Efficiency
Core
Own end-to-end performance optimization initiatives for high-impact production ML workloads, diagnosing bottlenecks and implementing improvements in inference execution, runtimes, and infrastructure.
Role type
Senior IC machine learning performance engineer
Builds
Faster, more scalable, and cost-efficient ML workloads for Reddit's Ads ML teams
Domain
Internet / Machine Learning Infrastructure
Deliverable
production ML models
Required skills
ML performance engineering, profiling and benchmarking, Python, systems programming, production ML frameworks, hardware-aware optimization
Preferred skills
PyTorch optimization (Dynamo, Inductor, AOTI), inference runtimes (TensorRT, ONNX Runtime, CUDA, Triton), distributed training, custom kernel development
Technologies
PyTorch, TensorRT, ONNX Runtime, CUDA, Triton, LLVM/MLIR, Dynamo, Inductor, AOTI
Responsibilities
Profile and diagnose ML performance bottlenecks across model execution, frameworks, runtimes, memory, and hardware; Design and implement performance improvements involving inference execution, runtime behavior, batching, and memory management; Improve ML training and serving efficiency and scalability; Support faster and safer model launches through benchmarking and load testing; Build reusable performance capabilities including benchmarks and tooling; Contribute to technical direction and mentor engineers
Seniority
Senior, hands-on IC
