ML Systems Engineer, Offline Infrastructure & Pipelines

Unity Technologies; Bellevue (WA); On-site; USD 92,000 - 138,000; Full time

Benefits offered by this job

Health insurance

Commuting subsidy

Employee stock ownership

Retirement plans

Generous vacation

Parental leave support

Office snacks

Mental health programs

ERGs

Training and development

Job summary

Unity Technologies in Bellevue, WA is seeking a Machine Learning Engineer to join our Offline Infrastructure team. You will help build data pipelines, support distributed training workflows, and improve reproducibility of ML pipelines at scale.

This role bridges research and production—working with engineers and researchers to translate advanced ideas into reliable systems for training data generation and model iteration.

Qualifications

• Bachelor’s degree in Computer Science, ML, Systems, or related field.
• Experience with Python and data-intensive workloads.
• Familiarity with ML frameworks (PyTorch, TensorFlow) and/or distributed systems (Ray, Spark).
• Experience with data pipelines, model training workflows, or large datasets.

Responsibilities

• Build and maintain data pipelines that generate training datasets for ML models and experiments.
• Contribute to infrastructure that supports distributed training workflows (e.g., PyTorch, Ray).
• Work with workflow orchestration tools (Airflow, Flyte) to support multi-stage ML pipelines.
• Improve reproducibility and reliability through dataset validation, monitoring, and testing.
• Partner with ML engineers to support experimentation and model iteration.
• Help optimize performance and efficiency across data processing and training systems.
• Contribute to the evolution of offline ML platform architecture as it scales.

Skills

ML systems

Distributed systems

Python for data workloads

Model training workflows

Problem solving

Scalable infrastructure

Research to production translation

Education

Bachelor’s degree in CS / ML / Systems

Tools

PyTorch

Ray

Airflow

Flyte

Spark

TensorFlow

Job description

Unity Technologies in Bellevue, WA is seeking a Machine Learning Engineer to join our Offline Infrastructure team. You will help build data pipelines, support distributed training workflows, and improve reproducibility of ML pipelines at scale.

This role bridges research and production—working with engineers and researchers to translate advanced ideas into reliable systems for training data generation and model iteration.