Staff/ Senior Machine Learning Engineer, Data Infrastructure

Unity Software
Shanghai
Workplace: OnsiteFull timeFunction: Data Science & Machine LearningSkills: ["Systems thinking","Technical direction","Architectural influence"]

Design and evolve a large-scale offline data platform that generates training datasets and orchestrates reliable multi-stage ML workflows. Build batch and streaming data infrastructure using distributed systems, integrate with orchestration tools, and improve reproducibility via validation, monitoring, and automated testing. Optimize distributed compute performance and cost, partner with ML engineers on experimentation and model iteration, and lead architectural improvements for scalability and reliability.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
Unity Software
Unity Software
2 months ago

Staff/ Senior Machine Learning Engineer, Data Infrastructure

✓ Verified Job

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

Source: Company careers pageValidated by: Fursa AI
Last checked: 57 minutes agoStatus: Live

Job Summary

Design and evolve a large-scale offline data platform that generates training datasets and orchestrates reliable multi-stage ML workflows. Build batch and streaming data infrastructure using distributed systems, integrate with orchestration tools, and improve reproducibility via validation, monitoring, and automated testing. Optimize distributed compute performance and cost, partner with ML engineers on experimentation and model iteration, and lead architectural improvements for scalability and reliability.
Location: Shanghai
Workplace: Onsite
Employment Type: Full time
Job Function: Data Science & Machine Learning
Seniority: Mid level

Key Responsibilities

  • •Develop infrastructure supporting both batch and streaming big data processing.
  • •Design and operate large-scale data pipelines that generate training datasets for machine learning training and experimentation.
  • •Integrate data pipelines with workflow orchestration systems (e.g., Flyte, Airflow) to enable reliable multi-stage training workflows.
  • •Improve reproducibility and observability of data pipelines through dataset validation, monitoring, and automated testing.
  • •Optimize performance and resource utilization across distributed compute systems and lead architectural improvements to keep offline pipelines scalable, reliable, and cost-efficient.

Pay and Benefits

Equity and Bonus:Equity
Perks:Health InsuranceLife InsuranceDisability InsuranceCommuter BenefitsEquityRetirementPaid LeaveParental Leave

Key Requirements

  • •Experience working with distributed computing frameworks such as Flink, Spark, and Ray for distributed data processing.
  • •Experience building infrastructure for training data generation, dataset preparation, or ML feature pipelines.
  • •Experience optimizing big data pipelines with a focus on cost efficiency.
  • •Strong programming skills in Python and experience working with large-scale distributed workloads.
  • •Experience with modern data infrastructure such as data lakes, warehouses, orchestration systems, and streaming platforms.
Skills:Systems thinkingTechnical directionArchitectural influence
Languages:English
Tech Stack:FlinkSparkRayFlyteAirflowPython

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