Machine Learning Systems Engineer, Ads ML Platform

About the role

This position focuses on developing the foundational infrastructure that powers Reddit's advertising machine learning capabilities. You will contribute to building systems that enable efficient creation, management, and utilization of features and training datasets for ML models. The role is ideal for someone who enjoys constructing reliable data systems and developer tools rather than pure model development.

Key facts

What you'll do

• Construct data infrastructure to handle large-scale computation, transformation, and storage of features and training sets.
• Create frameworks for batch and real-time feature processing, prioritizing dependability, scalability, and user-friendliness.
• Implement platform features for managing features, including tracking their origin, validating them, detecting changes, monitoring for unusual patterns, ensuring reproducibility, and managing versions.
• Collaborate with ML engineers to integrate feature engineering processes into production ML systems.
• Develop systems that support automated ML processes, such as discovering features, assessing their quality, and managing their lifecycle.
• Enhance operational efficiency through improved observability, performance tuning, reliability engineering, and cost management.

Requirements

• A minimum of 3 years of experience in data infrastructure, platform engineering, or ML infrastructure.
• Proven experience building production services, data pipelines, APIs, workflow systems, or tools for developers.
• Hands-on familiarity with at least one distributed data or compute system, such as Spark, PySpark, Flink, Kafka, Ray, Airflow, Kubernetes, or BigQuery.
• Understanding of ML data workflows, including feature creation, training data generation, batch and real-time processing, model training, experimentation, or online serving.
• Proficient coding skills with the ability to produce clean, maintainable, and well-tested code.

Nice to have

• Experience building automated or agent-based workflows for ML systems.
• Familiarity with MLOps workflows, covering feature engineering, training pipelines, experimentation, model deployment, and online serving.

Skills & tools

• Spark
• PySpark
• Flink
• Kafka
• Ray
• Airflow
• Kubernetes
• BigQuery

Practical notes

Reddit offers global benefits programs that support various aspects of life, including workspace, professional growth, and caregiving. Specific benefits include family planning support, gender-affirming care, mental health and coaching services, a group personal pension scheme with employer matching, private medical and dental insurance, income replacement programs, a bike-to-work scheme, flexible vacation and paid volunteer time off, and generous paid parental leave. Interviews may be recorded, transcribed, and summarized by AI, with an option to opt out. Personal information collected during interviews will be used for evaluation purposes and will not be sold or shared for marketing. Recordings will be deleted promptly after a hiring decision. Reddit is an equal opportunity employer and provides reasonable accommodations for individuals with disabilities.