RemoteLeaf Logo
Log in Sign up

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

2 months ago
United States
Full-time
Entry Level
Machine Learning Engineer
Software Development
Apache Airflow Apache Spark Machine Learning Python PyTorch TensorFlow
Apply Now
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
View All Jobs 125

Description

  • Build and maintain data pipelines that generate training datasets for machine learning models and experimentation.
  • Contribute to infrastructure that supports distributed training workflows such as PyTorch and Ray.
  • Work with workflow orchestration tools such as Airflow, Flyte, or similar systems 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.
  • Strong English communication skills for professional verbal and written exchanges are required.
  • Relocation support is not available for this position.
  • Work visa or immigration sponsorship is not available for this position.

Benefits

  • Gross pay salary range of $112,700 to $169,100 USD.
  • Comprehensive health, life, and disability insurance.
  • Employee stock ownership.
  • 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.
  • 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

Stripe

Machine Learning Engineering Manager - Fraud Detection

Stripe 5K-10K Diversified Financial Services

Stripe is seeking an engineering leader to lead its Risk Detection team in building merchant-facing risk management experiences that reduce friction, protect users, and support the company’s growth.

Anywhere Full-time Senior Engineering Manager Machine Learning Engineer
Machine Learning
9 minutes ago
W

Staff Machine Learning Engineer, DUE

Waymo Autonomous vehicles, robotics, AI, ride-hailing / mobility tech

Waymo is hiring an AI Engineer for its Driver Understanding and Evaluation team to develop machine learning systems that evaluate autonomous driving behavior and improve how the Waymo Driver is assessed at scale.

United States Full-time Senior Machine Learning Engineer
$238k-$302k
C++ Deep Learning LLM Machine Learning MLOps Python PyTorch Reinforcement Learning TensorFlow Transformers
9 minutes ago
Plain Concepts

AI/ML Engineer (GenAI)

Plain Concepts 251-1K Internet Software & Services

Plain Concepts is hiring a Data Scientist to join its international AI/ML team and deliver custom AI solutions for client projects from development through production.

Spain Full-time Senior Machine Learning Engineer
Agile Azure CI/CD Computer Vision Docker gRPC Machine Learning MLOps NLP Python REST API SQL
24 minutes ago
3Pillar Global

Associate Architect

3Pillar Global 1K-5K Internet Software & Services

3Pillar is seeking an Architect/Principal, AI ML to define and lead the architecture for scalable enterprise AI solutions across client projects in a global, remote-first consulting environment.

Canada Full-time Lead Machine Learning Engineer Software Architect
Docker Generative AI Kubeflow Kubernetes Machine Learning Microservices MLflow MLOps Python PyTorch SageMaker Scikit-learn TensorFlow XGBoost
24 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