Staff Machine Learning Systems Engineer

• The Machine Learning Platform team at Reddit is a high-impact team that owns the infrastructure that powers recommendations, content discovery, user and content quantification, while directly impacting other teams such as Growth, Ads, Feeds, and Core Machine Learning teams
• As a Staff ML Infrastructure Engineer, you will lead development of a platform for large scale ML models at Reddit
• Design end-to-end model lifecycle patterns (MLOps) to boost velocity of development for ML engineers, including data preparation, model management, experiment tracking, and more
• Zero-to-one development and support of a graph ML codebase and platform that abstracts away common patterns and enables greater model scalability and iteration
• Collaborate with ML engineers on performance tuning, including improving model training time, efficiency, and GPU training costs in a large, distributed ML training environment
• Optimize batch data processing within a data warehouse and with tools such as Apache Beam, Apache Spark, Ray Data, and more
• Architect pipelines to build and maintain massive graph data structures on the order of billions of nodes and tens of billions of edges

Benefits

• Comprehensive health benefits

• Flexible vacation & Reddit global days off

• Family planning funds & 4+ months paid parental leave

• Personal & professional development funds

• Paid volunteer time off

• Workspace & home office benefits

• Deep experience with cloud-based technologies for supporting an ML platform, including tools like GCP BigQuery, Google Cloud Storage, infrastructure-as-code (Terraform), and more

• Hands-on experience administering and integrating MLOps tools for experiment tracking, model serving, and model registries (e.g. MLflow or Wandb)

• Strong focus on scalability, reliability, performance, and ease of use. You are an undying advocate for platform users and have a deep intuition for the machine learning development lifecycle

• Strong organizational & communication skills

• Proficiency with the common programming languages and frameworks of ML, such as Python, PyTorch, Tensorflow, etc

• Experience working with graph neural networks (GNNs) and associated graph ML frameworks (PyTorch Geometric, Deep Graph Library) is a big plus

• Deep experience working with distributed training frameworks, including Ray and Kubernetes

• Hands-on experience with ML optimization, including memory and GPU profiling

• Experience working with graph databases (Neo4j, JanusGraph, TigerGraph) is a big plus

• 8+ years of experience in ML infrastructure, including model training and model deployments