Unity logo

Staff Machine Learning Engineer, ML Infrastructure - Offline

Unity
Shanghai, China
Posted 3 hours ago
Last seen 3 hours ago
Active
full-time
AI & Machine Learning

Job Summary

Join Unity as a Staff Machine Learning Engineer to design and evolve a large-scale offline ML platform, focusing on building reliable infrastructure for training datasets and distributed model training. This role offers the chance to work with cutting-edge technologies and lead architectural improvements in a dynamic environment.

About this job

Unity Vector is building an offline ML platform for insights, experimentation, attribution, and AI-driven decision-making. This role involves designing and evolving a large-scale offline platform to build reliable infrastructure for generating training datasets, orchestrating ML workflows, and enabling efficient, distributed model training at scale.

Requirements

- Strong experience building large-scale ML pipelines - Experience working with distributed computing frameworks such as Ray, Spark, Flink and familiarity in the Ray ecosystem (Ray Data, Ray Train) for distributed data processing and model training - Experience building infrastructure for training data generation, dataset preparation, or ML feature pipelines - Deep experience designing and operating production-grade data pipelines - Strong programming skills in Python and experience working with large-scale distributed workloads - Experience with modern data infrastructure (data lakes, warehouses, orchestration systems, streaming platforms) - Strong systems thinking, with the ability to reason about performance, scalability, reliability, and cost tradeoffs in distributed systems - Proven ability to lead technical direction and influence architectural decisions across teams without formal authority

Benefits & Perks

- Comprehensive health, life, and disability insurance - Commute subsidy - Employee stock ownership - Competitive retirement/pension plans - Generous vacation and personal days - Support for new parents through leave and family-care programs - Office food snacks - Mental Health and Wellbeing programs and support - Employee Resource Groups - Global Employee Assistance Program - Training and development programs - Volunteering and donation matching program

Apply for this position

Apply Now

You'll be redirected to the company's application page to complete your application.

ManaBoard LogoManaBoard.io

The #1 platform for finding high-quality job postings in the gaming industry. Connect with top studios and talent.

Stay Updated

Get gaming job alerts and industry insights delivered to your inbox.

By subscribing, you agree to receive our newsletter and occasional updates. You can unsubscribe at any time.

Disclaimer: ManaBoard is an independent platform. Job listings and logos are sourced from public career pages and remain the property of their respective owners.

© 2026 ManaBoard. All rights reserved.