Staff Machine Learning Engineer, Offline Infrastructure

Unity

San Francisco, CA, United States
Base: $209,700 - $283,800 usd; bonus/equity: not s...
On-site
Strong experience building large-scale ml pipelines
Experience with ray data and ray train frameworks
Deep experience designing production-grade data pipelines
Unity Vector builds an offline ML platform that powers insight, experimentation, attribution, and AI-driven decision-making across the company

Job Summary

  • Unity Vector builds an offline ML platform that powers insight, experimentation, attribution, and AI-driven decision-making across the company.
  • The role focuses on building reliable infrastructure for generating training datasets, orchestrating ML workflows, and enabling efficient distributed model training at scale.
  • Unity offers comprehensive health insurance, employee stock ownership, competitive retirement plans, and generous vacation days to support team well-being.

Matching Summary

Unity Vector builds an offline ML platform that powers insight, experimentation, attribution, and AI-driven decision-making across the company.

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

  • 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 for performance and cost tradeoffs

Key Requirements

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

Work Rights

Not specified

Tailored Resume

Cover Letter