Senior Machine Learning Engineer, ML Infrastructure - Offline
Shanghai, China
Posted 3 hours ago
Last seen 3 hours ago
Active
full-time
AI & Machine Learning
Job Summary
Join Unity's AI & Machine Learning team as a Senior ML Engineer to shape the future of large-scale offline ML infrastructure. You'll design and optimize robust data pipelines and distributed training systems, powering AI-driven decision-making across Unity's global operations.
About this job
Unity Vector builds an offline ML platform that powers insight, experimentation, attribution, and AI-driven decision-making. This Senior ML Engineer role focuses on designing and evolving this large-scale offline platform, building reliable infrastructure for generating training datasets, orchestrating ML workflows, and enabling efficient, distributed model training at scale. You will ensure ML pipelines remain reliable, scalable, and architecturally sound.
Requirements
- 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 and optimizing large-scale distributed ML training pipelines with Torch Compilation, Quantization, CUDA, GPU kernel optimization etc.
- 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
You'll be redirected to the company's application page to complete your application.
Related Jobs
Similar positions you might be interested in
Explore Categories
Browse jobs by department and skill area