Unity

Staff Machine Learning Engineer

Unity
Apply
14 days ago
Mountain View, CA, USAStaff+
H1B sponsor

Base Salary

$172k - $284k/yr

Responsibilities

  • Set the technical vision and architecture for the ML platform from data ingestion through model serving.
  • Build reproducible and auditable data, model, prompt-versioning, evaluation, and cost-tracking systems.
  • Lead engineering practices for agentic development workflows and evaluation methodologies.
  • Mentor engineers and influence technical direction across disciplines and teams.
  • Build predictors for user acquisition cost, lifetime value, return on ad spend, retention, and related commercial outcomes.
  • Develop market-demand forecasting models and calibrated prediction ranges with confidence estimates.
  • Own ranking, portfolio, experimentation, feedback-loop, bias-correction, and measurement systems.
  • Build and operate LLM-agent evaluation harnesses, automated playtesting, multimodal content assessment, and production agent orchestration.

Requirements

  • Track record of owning production machine learning systems end to end from data through serving and measured business outcomes.
  • Deep experience in performance marketing or ad-tech modeling, including acquisition cost, lifetime value, return on ad spend, or retention prediction.
  • Strong knowledge of probabilistic prediction, calibration, quantile or interval prediction, coverage, bias correction, and forecast backtesting.
  • Experience building production LLM-based agent systems, including orchestration, evaluation harnesses, prompt and model versioning, and cost control.
  • Strong Python and modern machine learning tooling experience, with fundamentals across relational stores, columnar analytics stores, and event streams.
  • Ability to drive technical direction across teams, mentor senior engineers, and influence without formal authority.
  • Experience in a zero-to-one environment.
  • Preferred experience includes mobile game user acquisition, LiveOps, monetization, Unity or another game engine, automated playtesting, time-series forecasting, demand forecasting, Claude Agent SDK, Codex, BigQuery, ClickHouse, Kafka, Postgres, pgvector, forecasting, causal inference, calibration, or agentic-systems publications or open-source contributions.

Benefits

  • Comprehensive health, life, and disability insurance is available, with eligibility and coverage varying by country and employment status.
  • Benefits include commute subsidy, employee stock ownership, retirement or pension plans, vacation and personal days, parental and family-care leave, office food and snacks, mental health programs, employee resource groups, employee assistance, training and development, and volunteering and donation matching.
  • The posting lists a base salary range by work location: Zone A $218,400-$283,900, Zone B $194,100-$252,300, and Zone C $172,200-$223,900.
  • The role may be eligible for equity awards and company incentive plans, including annual discretionary bonuses or sales commissions.
  • The position requires professional English communication with worldwide colleagues and partners.

Tech Stack

Apache KafkaClickHouseGoogle BigQueryPostgreSQLPythonUnity
Unity

About Unity

5,001-10,000 employees

Unity is a public software company that builds the Unity real-time 3D engine and services for creating, operating, and monetizing games and interactive 3D content across mobile, PC, console, and XR. It sells subscription licenses and usage-based services such as cloud build, ads, and analytics to studios and enterprises in gaming, automotive, film, and AR/VR. Founded in 2004 and headquartered in San Francisco, it trades on the NYSE (U).

Contact me