Machine Learning Engineer, Ads Training Platform - Remote

Overview

Reddit is looking for a Machine Learning Engineer to design and maintain large-scale distributed training infrastructure for its Ads ML models, enabling fast and reliable model training across large datasets.

In Short

• Design and maintain distributed training infrastructure.
• Develop tools on the Ray platform.
• Debug and profile distributed training jobs.
• Improve data access patterns with object storage integration.
• Collaborate with ML engineers to optimize training efficiency.
• Enhance scheduling and fault tolerance in the training platform.

Requirements

• 3+ years in infrastructure/platform engineering.
• 2+ years experience with the Ray platform.
• Strong understanding of distributed computing principles.
• Experience with distributed storage systems.
• Proven debugging skills for distributed jobs.
• Experience with deep learning frameworks like PyTorch or TensorFlow is a plus.
• Bonus: experience in model optimization for distributed training.

Benefits

• Comprehensive Healthcare Benefits.
• 401k Match.
• Family Planning Support.
• Gender-Affirming Care.
• Mental Health & Coaching Benefits.
• Flexible Vacation & Global Days off.
• Generous paid Parental Leave.
• Paid Volunteer time off.