Senior Machine Learning Engineer, Ads Training Platform - Remote

Overview

Reddit is a community of communities. It’s built on shared interests, passion, and trust and is home to the most open and authentic conversations on the internet. Every day, Reddit users submit, vote, and comment on the topics they care most about. With 100,000+ active communities and approximately 101M+ daily active unique visitors, Reddit is one of the internet’s largest sources of information.

In Short

• Design, build, and maintain large-scale distributed training infrastructure for Ads ML models.
• Develop tools and frameworks on top of the Ray platform.
• Build tools to debug, profile, and tune distributed training jobs for performance and reliability.
• Integrate with object storage systems and improve data access patterns.
• Collaborate with ML engineers to improve model training time, efficiency, and GPU training costs.
• Drive improvements in scheduling, state management, and fault tolerance within the training platform.

Requirements

• 5+ years in infrastructure/platform engineering or large-scale distributed systems.
• 2+ years hands-on experience with Ray platform.
• Strong understanding of distributed computing principles.
• Experience with distributed storage systems and large-scale data processing.
• Proven ability to debug and profile distributed jobs.
• Experience with deep learning frameworks (PyTorch, TensorFlow) is a big plus.
• Bonus: model optimization for distributed training, Ads ML experience.

Benefits

• Private Pension plan with Employer-matching
• 100% employer-sponsored group medical plan
• Income Replacement Programs
• Family Planning Support
• Gender-Affirming Care
• Mental Health & Coaching Benefits
• Flexible Vacation & Reddit Global Days Off