Staff Machine Learning Systems Engineer

Reddit

Confirmed live 2 days ago High trust
Remote

Quick summary

Work type
Remote
Location
Remote
Salary
$230,000–$322,000 / yr
Posted
177 days ago
Freshness
Confirmed live 2 days ago

Market check

Salary context

Above market

How this pay compares to similar roles

Similar $227k
This role $276k
$168k most similar roles pay here $338k

This role pays more than 85% of similar roles. Most pay $198,375–$254,750 — the shaded band above. At the midpoint, this role pays about $276k versus about $227k for comparable roles.

Based on 240 similar postings.

Employer

About Reddit

Reddit is a social news aggregation and discussion platform where users share content, vote on posts, and engage in community conversations across thousands of interest-based forums called subreddits.

Reddit currently has 77 open roles on FindRole.

Listed pay typically runs $217,000–$303,400 across 77 roles with salary data.

Most-posted roles

View all roles at Reddit

At a glance

TL;DR · Staff Machine Learning Systems Engineer

Staff Machine Learning Systems Engineer As part of the Machine Learning Platform team, the Staff Machine Learning Systems Engineer will lead the development of a platform for large-scale machine learning models. The role involves designing end-to-end MLOps patterns to improve developer velocity, building and supporting a graph ML codebase that abstracts common patterns, and collaborating on performance tuning to optimize training time and GPU costs in distributed environments. Key responsibilities include optimizing batch data processing and architecting pipelines for massive graph data structures. The candidate will utilize technologies including Python, PyTorch, TensorFlow, Ray, Kubernetes, Terraform, and GCP BigQuery. Experience with MLOps tools like MLflow or Wandb is required. Additionally, the role involves solving complex problems in content discovery and recommendation systems by managing large-scale infrastructure for model management, experiment tracking, and distributed training across billions of nodes.

What you'll do

  • Design end-to-end MLOps patterns including data preparation, model management, and experiment tracking.
  • Develop and support a graph ML codebase that abstracts common patterns for better scalability.
  • Perform performance tuning to improve training time, efficiency, and GPU costs in distributed environments.
  • Optimize batch data processing using tools like Apache Beam, Spark, and Ray Data.
  • Architect pipelines to build and maintain massive graph data structures with billions of nodes and edges.
  • Integrate MLOps tools for experiment tracking, model serving, and model registries.

What we're looking for

  • 8+ years of experience in ML infrastructure, including model training and deployment.
  • Hands-on experience with ML optimization, specifically memory and GPU profiling.
  • Deep experience with cloud technologies like GCP BigQuery, Google Cloud Storage, and Terraform.
  • Experience administering MLOps tools for experiment tracking, model serving, and registries like MLflow or Wandb.
  • Proficiency in common ML programming languages and frameworks such as Python, PyTorch, and TensorFlow.
  • Deep experience with distributed training frameworks including Ray and Kubernetes.
  • Ability to optimize batch data processing using tools like Apache Beam, Spark, or Ray Data.
  • Experience with graph databases (Neo4j, JanusGraph, TigerGraph) and graph neural networks is a plus.

More like this

Similar roles

Senior Machine Learning Systems Engineer

Reddit

Remote 177 days ago $216,700$303,400
MLOps Python PyTorch TensorFlow Kubernetes Ray Terraform Apache Beam Apache Spark GCP BigQuery Google Cloud Storage Neo4j JanusGraph TigerGraph PyTorch Geometric Deep Graph Library Distributed Training GPU Profiling
5+ yrs exp Remote

Staff Machine Learning Engineer

Intuit

Mountain View, CA 152 days ago $202,500$274,000
Python TensorFlow Keras Spark SQL NLTK NumPy Pandas AWS GCP Git GitHub Machine Learning Operations Data Pipelines R Data Structures Algorithms
7+ yrs exp

Staff Machine Learning Engineer

Intuit

Mountain View, CA 148 days ago $202,500$274,000
Python NLTK NumPy Pandas TensorFlow Keras R Spark SQL Git GitHub CUDA cuDNN AWS GCP Machine Learning Feature Engineering
6+ yrs exp

Staff Machine Learning Engineer

GEICO

Palo Alto, CA +1 98 days ago $130,000$260,000
Generative AI LLMs Python Java C++ C# Kubernetes CICD Kafka Spark Ray Airflow Temporal PostgreSQL MongoDB Cassandra Snowflake elastic search Qdrant
6+ yrs exp

Staff Machine Learning Engineer

Adobe

San Jose, CA 77 days ago $211,800$306,625
Python PyTorch TensorFlow JAX Hugging Face Ray Spark LLMs multimodal foundation models Vector Search Information Retrieval MLOps Distributed Training A/B Testing Transformer Architectures Representation Learning Metric Learning search, ranking

Staff Machine Learning Engineer

Arm Holdings

Austin, TX 32 days ago $249,900$338,100
Machine Learning Python TensorFlow PyTorch ARM Architecture GPU Neural Networks Computer Architecture hardware aware optimization Deep Learning
Hybrid