Unity

Machine Learning Engineer, New Grad

Unity
Apply
7 days ago
Remote, Worldwide +2 moreEntry Level

Base Salary

$104k - $152k/yr

Responsibilities

  • Build and maintain data pipelines that generate training datasets for machine learning models and experimentation.
  • Contribute to infrastructure supporting distributed model training workflows.
  • Use workflow orchestration tools to support multi-stage machine learning pipelines.
  • Improve reproducibility and reliability through dataset validation, monitoring, and testing.
  • Partner with ML engineers on experimentation and model iteration.
  • Optimize performance and efficiency across data processing and training systems.
  • Contribute to the architecture and scaling of the offline ML platform.

Requirements

  • Bachelor’s degree 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 data-intensive workloads.
  • Familiarity with ML frameworks such as PyTorch or TensorFlow and/or distributed systems such as Ray or Spark.
  • Academic or applied experience with data pipelines, model training workflows, or large datasets.
  • Ability to translate research ideas into practical systems and interest in scalable, reliable ML infrastructure.
  • Preferred qualifications include experience with Airflow or Flyte, exposure to data lakes, warehouses, or streaming systems, and publications or research in ML systems or distributed systems.

Benefits

  • Comprehensive health, life, and disability insurance, with specific eligibility and coverage varying by country and employment status.
  • Commute subsidy, employee stock ownership, competitive retirement or pension plans, vacation and personal days, and new-parent and family-care support.
  • Office food and snacks, mental health and wellbeing programs, Employee Resource Groups, and a global employee assistance program.
  • Training and development programs plus volunteering and donation matching.
  • Relocation support and work visa or immigration sponsorship are not available.
  • The position’s work location may be Mountain View, the San Francisco Bay Area, Bellevue, Seattle, New York City, or Remote California.

Tech Stack

Apache AirflowApache SparkPythonPyTorchTensorFlow

Categories

Data EngineeringML Engineering
Unity

About Unity

5,001-10,000 employees

Unity [NYSE: U] offers a suite of tools to create, market, and grow games and interactive experiences across all major platforms from mobile, PC, and console, to extended reality. For more information, visit Unity.com. Download Unity: http://unity3d.com/get-unity We're hiring! Visit our careers site: https://careers.unity.com For support, please email: support@unity3d.com Check us out on our other social channels: Twitter: https://twitter.com/unity Facebook: https://www.facebook.com/unity3d/ Instagram: https://www.instagram.com/unitytechnologies/

Contact me