Staff Machine Learning Engineer, Ml 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 designing reliable infrastructure for generating training datasets, orchestrating ML workflows, and enabling efficient distributed model training at scale.
  • Benefits include comprehensive health insurance, employee stock ownership, competitive retirement plans, and generous vacation days.

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
  • Experience with modern data lakes and warehouses
  • Ability to lead technical direction without formal authority
  • Systems thinking regarding performance and cost tradeoffs

Key Requirements

  • Proven ability to lead technical direction and influence architectural decisions
  • Strong systems thinking with reasoning about scalability and reliability
  • Experience working with large-scale distributed workloads in Python

Work Rights

Not specified

Tailored Resume

Cover Letter