RemoteLeaf Logo
Log in Sign up

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

1 month, 2 weeks 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 124

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

Affirm

Software Engineer II, Machine Learning (Feature Platform)

Affirm 1K-5K Diversified Financial Services

Affirm is hiring a software engineer for its ML Feature Platform team to build the self-serve data and feature infrastructure that powers machine learning and decisioning across the company.

United States Full-time Junior Data Engineer Machine Learning Engineer
$146k-$225k
AWS Kotlin Kubernetes Machine Learning MySQL Python
18 hours, 28 minutes ago
CI&T

[Job-30069] AI Engineer Specialist, Brazil

CI&T 5K-10K Internet Software & Services

A CI&T is hiring an AI Engineer Specialist to work with a global energy client on building and evolving a corporate AI platform for generative AI, intelligent agents, and process automation.

Brazil Full-time Senior AI Engineer Machine Learning Engineer
Azure CI/CD Databricks Docker FastAPI Generative AI Git Grafana Kafka Kubernetes LLM Microservices MLflow Prometheus Python Redis REST API
18 hours, 58 minutes ago
W

Senior Staff Machine Learning Engineer, LLM/VLM Model Architecture & Optimization

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

Waymo is hiring a machine learning engineer to advance perception-focused large model systems for its autonomous driving platform, with an emphasis on integrating models efficiently into the Waymo Driver.

United States Full-time Senior Machine Learning Engineer
$298k-$368k
Computer Vision Deep Learning LLM Machine Learning PyTorch
1 day, 18 hours ago
Block

Senior Machine Learning Engineer, Risk Modeling

Block 10K-50K Capital Markets

Block is hiring Senior and Staff Machine Learning Engineers for its Risk Machine Learning organization to develop large-scale fraud, abuse, and risk detection systems across Cash App and Square.

United States Full-time Lead Machine Learning Engineer
Apache Airflow Apache Spark AWS CI/CD Deep Learning GCP Keras LLM Machine Learning MLflow Mode MySQL NLP NumPy Pandas Prefect Python PyTorch Reinforcement Learning Scikit-learn Snowflake Tableau TensorFlow Vertex AI XGBoost
1 day, 18 hours 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