Staff Machine Learning Engineer, Offline Infrastructure

Unity

Mountain View, CA, United States
Base: $209,700 - $283,800 usd; bonus/equity: not s...
**
Large-scale ml pipeline design
Distributed computing frameworks ray spark flink
Python programming for distributed workloads
** Unity is seeking a Staff Machine Learning Engineer to enhance their offline ML infrastructure, focusing on scalable data pipelines and distributed model training. The role requires extensive experience with ML pipelines and distributed computing frameworks. The position is based in Mountain View, CA, and offers a comprehensive benefits package. **

Job Summary

  • The role focuses on designing and evolving a large-scale offline platform that powers insight, experimentation, and AI-driven decision-making across the company.
  • Candidates will develop infrastructure supporting distributed training workflows using technologies such as Pytorch, Ray Data, and Ray Train.
  • Unity offers comprehensive benefits including health insurance, employee stock ownership, competitive retirement plans, and generous vacation days.

Matching Summary

Match Score: 75

** Unity is seeking a Staff Machine Learning Engineer to enhance their offline ML infrastructure, focusing on scalable data pipelines and distributed model training. The role requires extensive experience with ML pipelines and distributed computing frameworks. The position is based in Mountain View, CA, and offers a comprehensive benefits package. **

Salary

Base: $209,700 - $283,800 USD; Bonus/Equity: Not specified; Benefits: Comprehensive health, life, disability insurance, employee stock ownership, retirement plans

Skills & Requirements

Must-have

  • Large-scale ML pipeline design
  • Distributed computing frameworks Ray Spark Flink
  • Python programming for distributed workloads
  • ML infrastructure for training data generation
  • Workflow orchestration systems Flyte Airflow

Nice-to-have

  • Strong systems thinking for tradeoffs
  • Ability to lead without formal authority
  • Experience with streaming data platforms
  • Proven architectural improvement leadership

Key Requirements

  • Strong experience building large-scale ML pipelines
  • Deep experience designing production-grade data pipelines
  • Strong programming skills in Python

Work Rights

Not specified

Tailored Resume

Cover Letter