Staff ML Platform Engineer (MLOps & Graph ML)

Aimlroles; Northern (KY); Hybrid; USD 230,000 - 322,000; Full time

Benefits offered by this job

Equity / RSUs

Excellent health benefits

Paid time off

Job summary

Reddit is seeking a Staff ML Infrastructure Engineer to lead the development of a platform for large-scale ML models. You will design end-to-end ML lifecycle patterns, scale graph data structures, and collaborate on performance tuning in a distributed training environment.

You will work with cloud technologies, MLOps tools, and modern ML frameworks to improve model training time and efficiency, while ensuring reliability and scalability for a broad set of users.

Qualifications

• Experience building ML infrastructure and platforms.
• Proficiency with cloud services and data engineering tools.
• ,

Responsibilities

• Lead development of a platform for large scale ML models.
• Design end-to-end MLOps patterns for model lifecycle, data prep, experiment tracking.
• Collaborate with ML engineers on performance tuning and GPU cost optimization.
• Optimize batch data processing within data warehouses and with tools like Apache Beam/Spark/Ray.
• Architect pipelines for massive graph data structures billions of nodes and edges.

Skills

ML infra

Cloud architecture

GPU profiling

Python

PyTorch

TensorFlow

Distributed training

MLOps

Performance optimization

Communication

Tools

GCP BigQuery

Google Cloud Storage

Terraform

MLflow

Wandb

Ray

Kubernetes

Neo4j

JanusGraph

TigerGraph

PyTorch Geometric

Deep Graph Library

Job description

Reddit is seeking a Staff ML Infrastructure Engineer to lead the development of a platform for large-scale ML models. You will design end-to-end ML lifecycle patterns, scale graph data structures, and collaborate on performance tuning in a distributed training environment.

You will work with cloud technologies, MLOps tools, and modern ML frameworks to improve model training time and efficiency, while ensuring reliability and scalability for a broad set of users.