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

1 week 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.
  • 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 is preferred.
  • Strong English communication skills for professional verbal and written exchanges are required.

Benefits

  • Gross pay salary range of $112,700 to $169,100 USD.
  • Comprehensive health, life, and disability insurance.
  • Competitive retirement or pension plans.
  • Generous vacation and personal days.
  • Support for new parents through leave and family-care programs.
  • Mental health and wellbeing programs and support.
  • Employee stock ownership.
  • Commute subsidy.
  • Training and development programs.
  • Volunteering and donation matching program.

Interested in this position?

Apply directly on the company website

Apply Now

Similar Roles

Senior Manager, Machine Learning

Twilio 5K-10K Diversified Telecommunication Services

Twilio is hiring a remote Senior Manager, Machine Learning to lead and build foundational platform capabilities for its Conversation Memory team, advancing conversational AI features for customer engagement applications.

AWS Azure GCP Machine Learning MLOps
57 minutes ago

Senior Software Engineer - AI Data Applications

Motional 1K-5K Automotive

Motional is hiring an experienced full stack engineer to join its AI Data Application team and help build data curation and search applications that accelerate ML dataset generation and support the company’s driverless future.

AWS gRPC Machine Learning Python React REST API Terraform TypeScript
2 hours, 49 minutes ago

Senior Staff Applied ML Engineer

Kaseya 1K-5K IT Services

Kaseya is hiring an Applied ML Engineer to work across multiple product teams building AI-powered features, automated workflows, and production machine learning systems for its IT management and cybersecurity platform.

Apache Spark Feature Engineering Kubeflow LLM Microservices MLflow MLOps Pandas Python PyTorch SageMaker SQL Vertex AI
4 hours, 28 minutes ago

Senior Machine Learning Engineer, Zeitgeist, Personalization

Spotify Media

Spotify is hiring a Senior Machine Learning Engineer for the Personalization team in New York to build AI-powered systems that understand cultural trends and improve personalized listening experiences for millions of users.

GCP Generative AI Java LLM Machine Learning NLP Python Scala
9 hours, 19 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