Staff Machine Learning Engineer, Offline Infrastructure

Unity Technologies

San Francisco, CA, United States
Base: $209,700 - $283,800 usd; bonus/equity: emplo...
On-site
Strong experience building large-scale ml pipelines
Experience with ray data and ray train frameworks
Deep experience designing production-grade data pipelines
The role focuses on building reliable infrastructure for generating training datasets and orchestrating ML workflows at scale

Job Summary

  • The role focuses on building reliable infrastructure for generating training datasets and orchestrating ML workflows at scale.
  • Candidates will design large-scale data pipelines that generate training datasets used for machine learning training and experimentation.
  • Unity offers comprehensive benefits including employee stock ownership, competitive retirement plans, and generous vacation time.

Matching Summary

The role focuses on building reliable infrastructure for generating training datasets and orchestrating ML workflows at scale.

Salary

Base: $209,700 - $283,800 USD; Bonus/Equity: Employee stock ownership mentioned; Benefits: Comprehensive health, life, disability insurance, commute subsidy, and more

Skills & Requirements

Must-have

  • Strong experience building large-scale ML pipelines
  • Experience with Ray Data and Ray Train frameworks
  • Deep experience designing production-grade data pipelines
  • Strong programming skills in Python for distributed workloads
  • Experience integrating ML pipelines with workflow orchestration systems

Nice-to-have

  • Familiarity with Spark or Flink distributed computing frameworks
  • Ability to lead technical direction without formal authority
  • Experience with modern data lakes and streaming platforms
  • Systems thinking regarding performance and cost tradeoffs

Key Requirements

  • Proven ability to lead technical direction across teams
  • Strong systems thinking for distributed system reliability
  • Deep experience with Pytorch and distributed compute systems

Work Rights

Not specified

Tailored Resume

Cover Letter