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

1 week, 3 days 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 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

Full-Stack AI Engineer

Pavago IT Services

Our client is hiring a Full-Stack AI Engineer to build and deploy production-ready AI applications that combine software engineering, machine learning, and cloud infrastructure across the full product lifecycle.

AWS Azure CI/CD Dagster Docker Elasticsearch FastAPI Flask GCP Git GitHub HIPAA Hugging Face Kubeflow Kubernetes LLM Machine Learning Microservices MLflow MLOps Next.js Node.js Prefect Prometheus Python PyTorch React REST API SageMaker Serverless Snowflake SQL TensorFlow TypeScript Vue.js
1 hour, 43 minutes ago

Senior AI Platform Engineer

Wellhub 1-10 Gas Utilities

Wellhub is hiring a Senior AI Platform Engineer in Brazil to help build and evolve the cloud-native ML development platform that enables engineers and data scientists to develop and deploy AI at scale.

Apache Spark AWS CI/CD Kubeflow Kubernetes MLOps Python Terraform
17 hours, 54 minutes ago

Senior Software Engineer (Typescript / FrontEnd) - AI/ML

ClickHouse 51-250 IT Services

ClickHouse is hiring a Senior Software Engineer to build AI/ML-powered features for ClickHouse Cloud, connecting its high-performance database platform with end-to-end AI integrations and user-facing experiences.

AWS Azure ClickHouse GCP JavaScript Python React TypeScript
20 hours, 6 minutes ago

Senior/Staff Deep Reinforcement Learning Engineer

DoorDash 10K-50K Air Freight & Logistics

DoorDash is hiring a Senior/Staff Deep RL Engineer to develop and deploy real-time autonomous driving policies for its DD Labs team, from problem formulation and training through on-vehicle inference.

Deep Learning Reinforcement Learning
20 hours, 25 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