Unity
Senior Machine Learning Engineer, Data Infrastructure
$166K–$273K
Mountain View, CA, USA +1 · Senior · Full-time · Posted Sep 8, 2026
- Location
- Mountain View, CA, USA
Bellevue, WA, USA
About the job
Designs and leads scalable offline data infrastructure and pipelines that prepare machine-learning training datasets and support experimentation.
What you'll do
- Develop batch and streaming data infrastructure and operate pipelines for training dataset generation.
- Integrate workflow orchestration and improve pipeline reproducibility, observability, performance, and cost efficiency.
- Partner with ML engineers and lead architectural improvements across teams.
What they ask for
- Experience with distributed computing frameworks such as Flink, Spark, or Ray.
- Experience building training-data generation, dataset-preparation, or ML feature-pipeline infrastructure.
- Strong Python programming skills and experience with large-scale distributed workloads.
- Experience optimizing big data pipelines for cost efficiency and influencing technical direction across teams.
Pay
Zone A: $210,300 - $273,400 USD base salary; Zone B: $187,200 - $243,300 USD base salary; Zone C: $165,600 - $215,200 USD base salary
Tools
Flink, Spark, Ray, Flyte, Airflow, Python
AI focus
Builds data infrastructure and training datasets for machine-learning training, experimentation, and production ML workflows.
Model-written summary · the original posting is authoritative