Senior ML Platform Engineer (Graph & MLOps)

Reddit, Inc.; San Francisco (CA); On-site; USD 216,700 - 303,400; Full time

Benefits offered by this job

Equity (RSUs)

401(k) with employer match

Generous vacation time

Parental leave

Medical, dental, and vision insurance

Job summary

Reddit, Inc. is seeking a Senior ML Infrastructure Engineer to lead the development of a large-scale ML platform at Reddit. You will design end-to-end model lifecycle patterns, including data prep, model management, experiment tracking, and more.

You will collaborate with ML engineers to optimize training time and GPU costs in a distributed environment, and architect pipelines for billions of nodes and edges in graph data structures.

Qualifications

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

Responsibilities

• Design end-to-end ML platform patterns and enable scalable deployment of models.
• Collaborate with ML engineers on performance optimization and GPU training cost reduction.
• Build and maintain graph data pipelines and graph ML workflows.

Skills

ML infrastructure

Model training

GPU profiling

Cloud platforms

MLOps tools

Python

PyTorch

TensorFlow

Ray

Kubernetes

Apache Beam

Apache Spark

BigQuery

Terraform

Tools

MLflow

Wandb

PyTorch

TensorFlow

Ray

Kubernetes

Apache Beam

Apache Spark

BigQuery

Terraform

Job description

Reddit, Inc. is seeking a Senior ML Infrastructure Engineer to lead the development of a large-scale ML platform at Reddit. You will design end-to-end model lifecycle patterns, including data prep, model management, experiment tracking, and more.

You will collaborate with ML engineers to optimize training time and GPU costs in a distributed environment, and architect pipelines for billions of nodes and edges in graph data structures.