ML Infra Engineer, Offline Platform

Unity Technologies; San Francisco (CA); On-site; USD 92,000 - 138,000; Full time

Benefits offered by this job

Health insurance

Commute subsidy

Employee stock ownership

Retirement/pension plans

Generous vacation and personal days

Parental leave

Office snacks

Wellbeing programs

ERGs

EAP (Employee Assistance Program)

Training programs

Volunteering and donation matching

Job summary

Unity Vector, part of Unity Technologies, seeks a Machine Learning Engineer to join our Offline Infrastructure team in San Francisco. You will build data pipelines and distributed training systems that power ML workflows and experimentation.

You will work with PyTorch, Ray, Airflow, and other tools to scale model training, ensure reproducibility, and translate research ideas into production-ready infrastructure.

Qualifications

• Bachelor’s degree in CS, ML, systems or related field.
• Strong foundation in ML systems, distributed systems, or large-scale data processing.
• Experience with Python and data-intensive workloads.
• Familiarity with ML frameworks (PyTorch, TensorFlow) and distributed systems (Ray, Spark).
• Experience with data pipelines, model training workflows, or large datasets.
• Strong problem-solving skills and ability to translate ideas into practical systems.

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, or similar) 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 our offline ML platform architecture as it scales.

Skills

Python

ML systems

Distributed systems

Data processing

PyTorch

TensorFlow

Model training workflows

Data pipelines

Education

Bachelor’s degree in Computer Science, Machine Learning, Systems, or a related field

Tools

PyTorch

Ray

Spark

Airflow

Flyte

Job description

Unity Vector, part of Unity Technologies, seeks a Machine Learning Engineer to join our Offline Infrastructure team in San Francisco. You will build data pipelines and distributed training systems that power ML workflows and experimentation.

You will work with PyTorch, Ray, Airflow, and other tools to scale model training, ensure reproducibility, and translate research ideas into production-ready infrastructure.