Senior/Staff Machine Learning Engineer, Unity
Data Infrastructure
$210.3-273.4k
More information about location
Office located in Mountain View, CA
Game development platform
Be an early applicant
Game development platform
1001+ employees
Be an early applicant
$210.3-273.4k
More information about location
Office located in Mountain View, CA
1001+ employees
Company mission
To democratize game development and enable everyone to create rich interactive content.
Role
Who you are
- Experience working with distributed computing frameworks such as Flink, Spark, Ray for distributed data processing
- Experience building infrastructure for training data generation, dataset preparation, or ML feature pipelines
- Experience optimizing big data pipelines and infrastructure for cost efficiency
- Strong programming skills in Python and experience working with large-scale distributed workloads
- Experience with modern data infrastructure (data lakes, warehouses, orchestration systems, streaming platforms)
- Strong systems thinking, with the ability to reason about performance, scalability, reliability, and cost tradeoffs in distributed systems
- Proven ability to lead technical direction and influence architectural decisions across teams without formal authority
What the job involves
- We are seeking a senior data infra engineer to design and evolve the large-scale offline platform. This role focuses on building reliable infrastructure for generating data infrastructure, training datasets, and orchestrating data workflows. You will work closely with ML engineers and platform teams to ensure our pipelines can efficiently handle growing data volumes and increasingly complex training workloads
- You will play a key role in shaping how model datasets are prepared to ensure the reliability, scalability, and performance of our data platform
- Develop infrastructure that supports both batch and stream big data processing using technologies such as Flink, Spark, Ray, etc
- Design and operate large-scale data pipelines that generate training datasets used for machine learning training and experimentation
- Integrate data pipelines with workflow orchestration systems (e.g., Flyte, Airflow, or similar) to enable reliable multi-stage training workflows
- Improve reproducibility and observability of data pipelines through dataset validation, monitoring, and automated testing
- Optimize performance and resource utilization across distributed compute systems used for data processing
- Partner closely with ML engineers to enable efficient large-scale experimentation and model iteration
- Lead architectural improvements to ensure our offline data pipelines remain scalable, reliable, and cost-efficient
Our take
Unity is one of the biggest names in game development, powering everything from indie hits to blockbuster titles. Its real-time 3D engine gives developers and artists the tools to build rich, interactive worlds that run across mobile, PC, console and beyond.
But Unity's ambitions stretch far beyond gaming and over the past few years, Unity's technology has found a home in industries including automotive, film and television, where companies use it to create digital twins, virtual productions and immersive experiences. Its tools also help power augmented reality experiences on devices from companies such as Apple.
In 2021, Unity was used to create 72% of the top 1,000 mobile games and half of all games released across mobile, pc and console. Since its 2020 IPO, the company has continued to expand its platform, and with significant post-IPO funding secured in 2025, Unity now supports more than 20 platforms, operates in over 190 countries and territories, and is continuing to push real-time 3D technologies into new industries.
Freddie
Company Specialist at Welcome to the Jungle
Insights
4% employee growth in 12 months
Company
Funding (last 2 of 7 rounds)
May 2019
$125m
SERIES E
Jun 2018
$145m
SERIES D
Total funding: $893.5m
Company benefits
- Comprehensive health, life and disability insurance
- Employee stock ownership
- Generous vacation and personal days
- Office food and free snacks, lots of health options!
- Employee Resource Groups
- Training and development programs
- Commute subsidy
- Competitive retirement/pension plans
- Paid leave for new parents
- Mental Health and Wellbeing programs and support
- Global Employee Assistance Program
- Volunteering and donation matching program
Company HQ
Yerba Buena, San Francisco, CA
Leadership
Matthew Bromberg
(CEO & President)Matthew is currently on the Board of Directors for Bumble. Previously, they were a Senior Advisor at Blackstone and COO at Zynga. They also spent over 4 years at EA as a Group GM of Bioware and as an SVP of Strategy & Operations of EA Mobile.
Salary benchmarks
We don't have enough data yet to provide salary benchmarks for this role.
Submit your salary to help other candidates with crowdsourced salary estimates.
Share this job
View 23 more jobs at Unity