Machine Learning Engineer, Offline Infrastructure (Entry-Level / New Grad)

Unity3D is a leading game engine company that builds an offline ML platform for AI-driven decision-making. They are seeking a Machine Learning Engineer to join their Offline Infrastructure team, where the role involves building and maintaining data pipelines and contributing to distributed training workflows.

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