ML Platform Infra Engineer - Scalable ML Pipelines

Unity Technologies; United States; Remote; USD 113,000 - 169,000; Full time

Benefits offered by this job

Health insurance

Employee stock ownership

Retirement plans

Generous vacation and personal days

Training and development programs

Job summary

Unity Technologies seeks a Machine Learning Engineer for the Offline Infrastructure team. You will build data pipelines for training data, support distributed training workflows, and improve reproducibility across ML pipelines at scale.

You’ll work with Python, PyTorch/TensorFlow, and orchestration tools to translate research ideas into production-ready systems. This role bridges research and production in a fast-growing ML platform.

Qualifications

• Bachelor’s degree in Computer Science, Machine Learning, Systems, or a related field.
• Experience with machine learning systems, distributed systems, or large-scale data processing from research or projects.
• Experience with Python and data-intensive workloads.
• Familiarity with ML frameworks (PyTorch, TensorFlow) and/or distributed systems (Ray, Spark).
• Strong problem-solving skills and ability to translate research ideas into practical systems.

Responsibilities

• Build and maintain data pipelines that generate training datasets for machine learning models and experimentation.
• Contribute to infrastructure that supports distributed training workflows (e.g., PyTorch, Ray).
• Work with workflow orchestration tools (e.g., 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

Education

Bachelor's degree

Tools

PyTorch

TensorFlow

Ray

Spark

Job description

Unity Technologies seeks a Machine Learning Engineer for the Offline Infrastructure team. You will build data pipelines for training data, support distributed training workflows, and improve reproducibility across ML pipelines at scale.

You’ll work with Python, PyTorch/TensorFlow, and orchestration tools to translate research ideas into production-ready systems. This role bridges research and production in a fast-growing ML platform.