Job Title: Staff Machine Learning Engineer

Company Name: Unity

Job Url: https://jobright.ai/jobs/info/69baf0d53b74eb1e2c892ee8

Job Description: Unity
· 1 day ago
Staff Machine Learning Engineer

United States

Full-time

Remote

Senior Level, Lead/Staff

$178K/yr - $268K/yr
52%
FAIR MATCH
64%
Exp. Level
24%
Skill
75%
Industry Exp.
Maximize your interview chances
Unity3D is the world’s leading game engine, powering play for more than 3 billion consumers each month. They are seeking a senior ML engineer to design and evolve their large-scale offline platform, focusing on building reliable infrastructure for generating training datasets and enabling efficient, distributed model training at scale.
3D Printing
Gaming
Information Technology
Big Data
3D Technology
PC Games
Real Time
No H1B

Insider Connection @Unity
2 email credits available today

Discover valuable connections within the company who might provide insights and potential referrals.
Get 3x more responses when you reach out via email instead of LinkedIn.
Beyond Your Network
T
E
S
N
D
Tian Wang & 4 connections
From Your Previous Company
From Your School
Find Any Email


Responsibilities
Design and operate large-scale data pipelines that generate training datasets used for machine learning training and experimentation
Develop infrastructure that supports distributed training workflows using technologies such as Pytorch, Ray Data, and Ray Train, etc
Integrate ML pipelines with workflow orchestration systems (e.g., Flyte, Airflow, or similar) to enable reliable multi-stage training workflows
Improve reproducibility and observability of ML pipelines through dataset validation, monitoring, and automated testing
Optimize performance and resource utilization across distributed compute systems used for data processing and model training
Partner closely with ML engineers to enable efficient large-scale experimentation and model iteration
Lead architectural improvements to ensure our offline ML pi