ML Systems Engineer - Offline Infrastructure (PhD)

3M HEALTHCARE

San Francisco (CA)

On-site

USD 112,700 - 169,100

Full time

14 days+

Get more replies from employers

Send a job-specific resume in minutes.

Benefits offered by this job

Health insurance
Stock options
Retirement plan
Vacation & personal days
Parental leave
Commuter subsidy

Job summary

Unity Technologies, a leading real-time 3D development platform, is seeking a Machine Learning Engineer for its Offline Infrastructure team in San Francisco. You will help build and evolve data pipelines, support distributed model training, and translate research ideas into scalable systems.

Ideal candidates hold a PhD and have strong ML systems experience, Python proficiency, and familiarity with PyTorch/Ray/Spark.

Qualifications

  • PhD in Computer Science, ML, or related field.
  • Strong background in ML systems and distributed data processing.
  • Experience with Python for data-intensive workloads.
  • Familiarity with PyTorch, TensorFlow and distributed systems (Ray, Spark).
  • Experience with data pipelines and large datasets.
  • Ability to translate research ideas into production systems.

Responsibilities

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

Skills

PhD in CS/ML
Distributed systems
Python
PyTorch
Data pipelines
Large-scale data

Education

PhD in CS/ML/Systems

Tools

PyTorch
TensorFlow
Ray
Spark

Job description

Unity Technologies, a leading real-time 3D development platform, is seeking a Machine Learning Engineer for its Offline Infrastructure team in San Francisco. You will help build and evolve data pipelines, support distributed model training, and translate research ideas into scalable systems.

Ideal candidates hold a PhD and have strong ML systems experience, Python proficiency, and familiarity with PyTorch/Ray/Spark.

Get your free, confidential resume review.
or drag and drop your file here.
Similar jobs

Similar jobs worth comparing

Machine Learning Engineer, Offline Infrastructure (Entry-Level / New Grad PhD)
Machine Learning Engineer, Offline Infrastructure (Entry-Level / New Grad PhD)

3M HEALTHCARE • Mountain View (CA)

On-site
USD 112,000 - 163,000
Employee stock ownership
Commuter subsidy
Generous vacation and personal days
+1
Machine Learning Engineer, Offline Infrastructure (Entry-Level / New Grad PhD)
Machine Learning Engineer, Offline Infrastructure (Entry-Level / New Grad PhD)

3M HEALTHCARE • San Francisco (CA)

On-site
USD 112,000 - 170,000
Health insurance
Stock options
Retirement plan
+3
Offline ML Infrastructure Engineer for Scalable Pipelines
Offline ML Infrastructure Engineer for Scalable Pipelines

3M HEALTHCARE • Mountain View (CA)

On-site
USD 112,000 - 163,000
Employee stock ownership
Commuter subsidy
Generous vacation and personal days
+1
Senior ML Engineer, Data Infrastructure & Pipelines
Senior ML Engineer, Data Infrastructure & Pipelines

Unity Enterprise • Mountain View (CA)

Hybrid
USD 200,000 - 261,000
Comprehensive health insurance
Life and disability insurance
Employee stock ownership
+3
Senior ML Engineer, Data Infrastructure & Pipelines
Senior ML Engineer, Data Infrastructure & Pipelines

Unity Technologies • Mountain View (CA)

On-site
USD 200,000 - 261,000
Health insurance
Stock options
Retirement plan
+3
Senior ML Data Infrastructure Engineer — Scalable Pipelines
Senior ML Data Infrastructure Engineer — Scalable Pipelines

Unity • Mountain View (CA)

On-site
USD 200,000 - 261,000
Comprehensive health insurance
Employee stock ownership
Competitive retirement plans
+4
Senior/Staff Machine Learning Engineer, Data Infrastructure
Senior/Staff Machine Learning Engineer, Data Infrastructure

Unity • Mountain View (CA)

On-site
USD 200,000 - 261,000
Comprehensive health insurance
Employee stock ownership
Competitive retirement plans
+4
Senior Real-Time ML Infrastructure Engineer
Senior Real-Time ML Infrastructure Engineer

3M HEALTHCARE • Bellevue (WA)

Remote
USD 183,000 - 249,000
Senior/Staff Machine Learning Engineer, Data Infrastructure
Senior/Staff Machine Learning Engineer, Data Infrastructure

Unity Technologies • Mountain View (CA)

On-site
USD 200,000 - 261,000
Health insurance
Stock options
Retirement plan
+3
Remote Senior ML Infra Engineer — Real-Time Model Serving
Remote Senior ML Infra Engineer — Real-Time Model Serving

3M HEALTHCARE • Mountain View (CA)

On-site
USD 183,000 - 249,000
Health insurance
Stock options
Retirement plan
+1