Senior/Staff Machine Learning Engineer, Data Infrastructure
Shanghai, China
Posted 11 hours ago
Last seen 11 hours ago
Active
full-time
AI & Machine Learning
Job Summary
Join Unity as a Senior/Staff Machine Learning Engineer to design and evolve a large-scale offline data platform, powering AI-driven decision-making and supporting critical ML pipelines. Leverage your expertise in distributed computing and data infrastructure to build scalable and reliable systems for model training and experimentation.
About this job
Unity Vector builds a data platform powering insights, experimentation, attribution, and AI-driven decision-making across the company. This role focuses on designing and evolving a large-scale offline platform, building reliable infrastructure for generating data infrastructure, training datasets, and orchestrating data workflows. You will ensure ML pipelines remain reliable, scalable, and architecturally sound.
Requirements
- Experience working with distributed computing frameworks such as Flink, Spark, Ray for distributed data processing
- Experience building infrastructure for training data generation, dataset preparation, or ML feature pipelines
- Experience optimizing big data pipelines and infrastructure for cost efficiency
- 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