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

1 hour, 49 minutes ago
Full-time
Entry Level
Software Development
Unity

Unity

Unity is the top platform for real-time 3D content creation, empowering creators across industries to bring their ideas to life with interactive 2D and 3D content.

Internet Software & Services
5K-10K
Founded 2004

Description

  • Build and maintain data pipelines that generate training datasets for machine learning models and experimentation.
  • Contribute to infrastructure that supports distributed training workflows using tools such as PyTorch and Ray.
  • Work with workflow orchestration tools such as Airflow or Flyte 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 the offline ML platform architecture as it scales.

Requirements

  • 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 data-intensive workloads.
  • Familiarity with ML frameworks such as PyTorch or TensorFlow and/or distributed systems such as Ray or 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.
  • Experience with workflow orchestration systems such as Airflow or Flyte is preferred.
  • Exposure to large-scale data platforms such as data lakes, warehouses, or streaming systems is preferred.
  • Publications or research in ML systems, distributed systems, or related areas are preferred.
  • English proficiency for frequent professional verbal and written communication with global colleagues and partners.

Benefits

  • Gross pay salary of $112,700 to $169,100 USD.
  • Comprehensive health, life, and disability insurance.
  • Competitive retirement/pension plans and employee stock ownership.
  • Generous vacation and personal days.
  • Support for new parents through leave and family-care programs.
  • Mental health and wellbeing programs and support.
  • Training and development programs.
  • Commute subsidy and office food snacks.

Interested in this position?

Apply directly on the company website

Apply Now

Similar Roles

Manager, Machine Learning Engineering (Underwriting)

Affirm 1K-5K Diversified Financial Services

Affirm is hiring a Machine Learning Engineering Manager to lead an underwriting ML team building decisioning systems that optimize application outcomes and support the company’s broader machine learning strategy.

Deep Learning Machine Learning Transformers
4 hours, 33 minutes ago

Systems & AI Cloud Architect

Endeavour. Inspired Infrastructure. 11-50 Electric Utilities

Endeavour is seeking a remote Systems & AI Cloud Architect to support its IT ecosystem by shaping enterprise architecture, modernizing infrastructure, and enabling scalable AI and cloud solutions for sustainable infrastructure initiatives.

AWS Azure CI/CD Cybersecurity GCP Generative AI Machine Learning Microservices MLOps
6 hours, 33 minutes ago

Senior Machine Learning Engineer, Personalization, Magenta

Spotify Media

Spotify is hiring a Senior Machine Learning Engineer on the Personalization team to build production conversational and agentic AI systems that improve how hundreds of millions of listeners discover and engage with audio.

LLM Machine Learning NLP
18 hours, 26 minutes ago

Machine Learning Engineer

Mindera 1K-5K Internet Software & Services

Mindera is seeking an ML Engineer to work with the ML Architect on machine learning frameworks and platform tooling that support scalable model development, deployment, and experimentation across business units.

Agile Apache Airflow CI/CD Docker Kubernetes MLflow MLOps PyTorch Scikit-learn TensorFlow
19 hours, 55 minutes ago

You're on a roll! Sign up now to keep applying.

Sign Up

Already have an account? Log in

Used by 14,729+ remote workers