Senior ML Infrastructure Engineer

Unity Enterprise; Mountain View (CA); On-site; USD 179,000 - 232,000; Full time

Job summary

Unity Technologies is seeking a Sr. Machine Learning Engineer in Mountain View, CA, to help scale ML infrastructure and feature pipelines across the platform.

You will work on distributed training, model serving, and observability, partnering with researchers and engineers to push production ML forward. The role requires strong Python, ML frameworks (PyTorch), and data pipeline experience, with cloud and Kubernetes familiarity as beneficial components of a broad ML stack.

Qualifications

• Experience building ML infrastructure, data platforms, or distributed systems in production.

Responsibilities

• Build and maintain data and feature pipelines for training datasets.

Skills

Python

ML infrastructure

PyTorch

Ray

Spark

Data pipelines

Distributed systems

Kubernetes

Cloud platforms

Observability tools

Education

Bachelor's degree in CS/ML/Systems

Tools

Airflow

Prefect

KServe

Triton

Kafka

Prometheus

Grafana

Job description

Unity Technologies is seeking a Sr. Machine Learning Engineer in Mountain View, CA, to help scale ML infrastructure and feature pipelines across the platform.

You will work on distributed training, model serving, and observability, partnering with researchers and engineers to push production ML forward. The role requires strong Python, ML frameworks (PyTorch), and data pipeline experience, with cloud and Kubernetes familiarity as beneficial components of a broad ML stack.