Machine Learning Engineer, Offline Infrastructure… at Unity · San…

Summary: Unity3D is the world’s leading game engine, and they are seeking a Machine Learning Engineer to join their Offline Infrastructure team. This role is ideal for a recent PhD graduate who will contribute to building and evolving the infrastructure for machine learning workflows and large-scale model training. Responsibilities: - Build and maintain data pipelines that generate training datasets for machine learning models and experimentation - Contribute to infrastructure that supports distributed training workflows (e.g., PyTorch, Ray) - Work with workflow orchestration tools (e.g., Airflow, Flyte, or similar) to support multi-stage ML pipelines - Improve reproducibility and reliability through dataset validation, monitoring, and testing - Partner with ML engineers to support experimentation and model iteration - Help optimize performance and efficiency across data processing and training systems - Contribute to the evolution of our offline ML platform architecture as it scales Required Qualifications: - PhD in Computer Science, Machine Learning, Systems, or a related field - Strong foundation in machine learning systems, distributed systems, or large-scale data processing (through research or projects) - Experience with Python and working with data-intensive workloads - Familiarity with ML frameworks (e.g., PyTorch, TensorFlow) and/or distributed systems (e.g., Ray, Spark) - Experience (academic or applied) with data pipelines, model training workflows, or large datasets - Strong problem-solving skills and ability to translate research ideas into practical systems - Interest in building scalable, reliable infrastructure for machine learning Preferred Qualifications: - Experience with workflow orchestration systems (Airflow, Flyte, etc.) - Exposure to large-scale data platforms (data lakes, warehouses, streaming systems) - Publications or research in ML systems, distributed systems, or related areas Required Skills: Machine learning systems, Distributed systems, Large-scale data processing, Python, PyTorch, TensorFlow, Ray, Spark, Data pipelines, Model training workflows, Workflow orchestration tools, Airflow, Flyte Benefits: 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