Staff Machine Learning Engineer, Offline Infrastructure

Unity

New York, NY, 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.
  • Unity offers comprehensive benefits including 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
  • Proven ability to lead technical direction without formal authority
  • Experience with modern data lakes and streaming platforms
  • Ability to reason about performance and cost tradeoffs

Key Requirements

  • Strong experience building large-scale ML pipelines
  • Deep experience designing production-grade data pipelines
  • Strong programming skills in Python
  • Experience with distributed computing frameworks like Ray
  • Proven ability to lead technical direction

Work Rights

Not specified

Tailored Resume

Cover Letter