ML Data Infrastructure Engineer

AppLovin

Confirmed live yesterday High trust

Quick summary

Work type
On-site
Location
Palo Alto, CA
Salary
$221,000–$331,000 / yr
Posted
10 days ago
Freshness
Confirmed live yesterday

Market check

Salary context

Above market

How this pay compares to similar roles

Similar $198k
This role $276k
$129k most similar roles pay here $353k

This role pays more than 94% of similar roles. Most pay $154,468–$241,750 — the shaded band above. At the midpoint, this role pays about $276k versus about $198k for comparable roles.

Based on 240 similar postings.

Employer

About AppLovin

AppLovin enables businesses to advertise profitably with marketing technologies that attract customers, increase revenue, and track ad performance.

AppLovin currently has 7 open roles on FindRole.

Listed pay typically runs $153,000–$200,000 across 7 roles with salary data.

Most-posted roles

View all roles at AppLovin

At a glance

TL;DR · ML Data Infrastructure Engineer

As an ML Data Infrastructure Engineer on the ML Data Platform team, you will solve technical challenges by upgrading and implementing state-of-the-art software infrastructure within a high-performance, globally distributed ecosystem of services. You will design and build data processing infrastructure for model training and feature serving while optimizing for performance, reproducibility, and traceability. Your daily work involves collaborating with research teams to implement novel data processing architectures for emerging model paradigms and resolving performance bottlenecks across the training data pipeline from raw ingestion to feature delivery. You will also establish best practices and tooling for data infrastructure used across various machine learning teams. Required skills include experience with distributed computing frameworks like Apache Spark or Flink, strong software engineering fundamentals, systems design, and performance optimization to support high-throughput, fault-tolerant distributed systems.

What you'll do

  • Design and build data processing infrastructure for model training and feature serving.
  • Optimize infrastructure for performance, reproducibility, and traceability.
  • Develop novel data processing architectures for emerging model and training paradigms with research teams.
  • Identify and resolve performance bottlenecks across the entire training data pipeline.
  • Manage the flow of data from raw ingestion to final feature delivery.
  • Establish best practices and tooling for data infrastructure used across multiple ML teams.
  • Build high-throughput, fault-tolerant distributed systems using frameworks like Apache Spark or Flink.

What we're looking for

  • Must have 1 - 3 years of experience and a minimum of a BS and/or MS in Computer Science.
  • Strong software engineering fundamentals with experience building high-throughput, fault-tolerant distributed systems.
  • Hands-on experience with distributed computing frameworks such as Apache Spark or Flink.
  • Solid grounding in data structures, systems design, and performance optimization.
  • Strong problem-solving skills and attention to detail.
  • Background in MLOps, Data Infrastructure, or ML Infrastructure (preferred).
  • Experience with ML training pipelines, feature stores, or model-serving systems (preferred).

More like this

Similar roles

Staff ML Engineer, Ads ML Infrastructure

Apple Inc

New York, NY 88 days ago $184,700–$324,800
Machine Learning ML Serving ONNX Runtime TensorRT vLLM Ray GPU Kernels Distributed Systems Feature Stores Federated Learning Privacy-Preserving ML RPC Agile
8+ yrs exp

Staff ML Engineer, Ads ML Infrastructure

Apple Inc

Cupertino, CA 88 days ago $184,700–$324,800
Machine Learning ML Serving ONNX Runtime TensorRT vLLM Ray GPU Kernels Distributed Systems Feature Stores Federated Learning Privacy-Preserving ML RPC
8+ yrs exp

Staff ML Infrastructure Engineer

Apple Inc

Cupertino, CA 9 days ago $184,700–$324,800
Python Rust C++ Go PyTorch JAX TensorFlow Docker Kubernetes Spark Parquet Iceberg Lance Ray Data NVIDIA DALI WebDataset Mosaic StreamingDataset DuckDB Polars
10+ yrs exp

Staff Machine Learning Engineer, Ads ML Efficiency

Reddit

Remote (Ontario, Canada) 3 days ago $230,000–$322,000
Python Go C++ Rust Java PyTorch Distributed Ray TensorFlow Spark Distributed Systems Machine Learning Infrastructure Model Serving GPU Architecture Performance Engineering Profiling Data Pipelines Cloud Infrastructure Benchmarking
5+ yrs exp Remote

ML Data Operations Engineer

Apple Inc

Cupertino, CA 73 days ago $184,700–$324,800
Machine Learning Data Pipelines Annotation Tools Dataset Management Technical Documentation Data Schema User Research Operations Data Collection
10+ yrs exp

ML Data Operations Engineer

Apple Inc

Santa Clara, CA 45 days ago $150,400–$225,300
Machine Learning Data Pipelines Dataset Management Technical Documentation Data Schema User Research Operations Behavioral Research Data Collection