Staff ML Infrastructure Engineer - Remote

Overview

As a Staff ML Infrastructure Engineer at Reddit, you will lead the development of a platform for large scale ML models, impacting various teams and enhancing the machine learning development lifecycle.

In Short

• Design end-to-end model lifecycle patterns (MLOps) for ML engineers.
• Develop and support a graph ML codebase and platform.
• Collaborate on performance tuning in a distributed ML training environment.
• Optimize batch data processing with tools like Apache Beam and Spark.
• Architect pipelines for massive graph data structures.
• 7+ years of experience in ML infrastructure.
• Hands-on experience with ML optimization and cloud technologies.
• Experience with MLOps tools for experiment tracking and model serving.
• Proficient in common ML programming languages and frameworks.
• Strong organizational and communication skills.

Requirements

• 7+ years of experience in ML infrastructure.
• Hands-on experience with ML optimization.
• Deep experience with cloud technologies (GCP, Terraform).
• Experience with MLOps tools (MLflow, Wandb).
• Proficient in Python, PyTorch, Tensorflow.
• Experience with distributed training frameworks (Ray, Kubernetes).
• Strong focus on scalability and performance.
• Experience with graph databases is a plus.
• Experience with graph neural networks is a plus.
• Strong communication skills.

Benefits

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