Machine Learning Engineer, ML Infrastructure

Unity Software
Shanghai
Workplace: RemoteFull timeFunction: Data Science & Machine LearningSkills: ["Systems thinking","Technical ownership","Technical leadership"]

Build and evolve Unity’s large-scale offline ML infrastructure that generates training datasets, orchestrates multi-stage ML workflows, and supports distributed model training. Design reliable, scalable data pipelines integrated with orchestration systems to improve reproducibility, observability, and automated validation. Optimize performance and resource utilization across distributed compute while partnering with ML engineers to accelerate experimentation and architectural improvements.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
Unity Software
Unity Software
2 days ago

Machine Learning Engineer, ML Infrastructure

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 8 hours agoStatus: Live

Job Summary

Build and evolve Unity’s large-scale offline ML infrastructure that generates training datasets, orchestrates multi-stage ML workflows, and supports distributed model training. Design reliable, scalable data pipelines integrated with orchestration systems to improve reproducibility, observability, and automated validation. Optimize performance and resource utilization across distributed compute while partnering with ML engineers to accelerate experimentation and architectural improvements.
Location: Shanghai
Workplace: Remote
Employment Type: Full time
Job Function: Data Science & Machine Learning
Seniority: Mid level

Key Responsibilities

  • •Design and operate large-scale data pipelines that generate training datasets for ML training and experimentation.
  • •Develop distributed training infrastructure to support large-scale model training workflows.
  • •Integrate ML pipelines with workflow orchestration systems (e.g., Flyte or Airflow).
  • •Improve reproducibility, observability, and reliability via dataset validation, monitoring, and automated testing.
  • •Optimize distributed compute performance and resource utilization while partnering on experimentation and architectural improvements.

Pay and Benefits

Perks:Health InsuranceLife InsuranceDisability InsuranceCommuter BenefitsEquityPensionPaid LeaveMeal Allowance

Key Requirements

  • •Strong experience building large-scale ML pipelines.
  • •Experience with distributed computing frameworks such as Ray, Spark, and Flink, including Ray ecosystem (Ray Data, Ray Train).
  • •Experience building infrastructure for training data generation, dataset preparation, or ML feature pipelines.
  • •Deep experience designing and operating production-grade data pipelines.
  • •Strong Python skills for large-scale distributed workloads and modern data infrastructure (data lakes/warehouses, orchestration, streaming).
Skills:Systems thinkingTechnical ownershipTechnical leadership
Languages:English
Tech Stack:PythonPyTorchRay DataRay TrainRaySparkFlinkFlyteAirflow

Eligibility

Work Authorization:Authorization required. Sponsorship not provided.

Company Brief

Unity Software
Develops the Unity real-time 3D development platform used to create, operate, and monetize interactive experiences and games across mobile, console, PC, and AR/VR platforms. Serves developers, studios, and enterprises worldwide.
Industry: Developer Tools
Company Size: Enterprise (1,001+ employees)
Revenue: USD 1B+
Growth: Public Company
Valuation: Public Company (Market Cap in USD)
Funding: IPO / Publicly Listed
Headquarters: San Francisco, United States
Founded: 2004
Glassdoor
Glassdoor: 3.8
WebsiteLinkedIn