Integrate ML pipelines with workflow orchestration systems (e.g., Flyte, Airflow, or similar) to enable reliable multi-stage training workflows + Improve reproducibility and observability of ML pipelines through dataset validation, monitoring, and automated testing + Optimize performance and resource utilization across distributed compute systems used for data processing and model training + Partner closely with ML engineers to enable efficient large-scale experimentation and model iteration + Lead architectural improvements to ensure our offline ML pipelines remain scalable, reliable, and cost-efficient **What we're looking for** + 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 **Additional information** + Relocation support is not available for this position **Benefits** At Unity, we want our team members to thrive. While specific benefits vary, here are some of the ways we strive to take care of our eligible team members globally: 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 **Life at Unity** Unity [NYSE: U] is the world's leading game engine, powering play for more than 3 billion consumers each month.