Unity3d

Unity3d

ML Infrastructure Engineer - (Early Career/Internship)

Remote, Washington, USA · Intern · Internship

No sponsorship$113k-$169kDetected 23 days ago
PythonDistributed SystemsMachine LearningTensorFlowPyTorchSparkAirflowData EngineeringMLOpsAI OrchestrationA/B TestingUnityAR/VRResearchCommunicationProblem Solving

About the role

  • Our systems operate at scale across batch and streaming data, supporting analytics, product intelligence, machine learning pipelines, and business operations.
  • As data volume and complexity grow, our platform enables large-scale model training, feature generation, and experimentation workflows that power production ML systems.
  • We're looking for a Machine Learning Engineer to join our Offline Infrastructure team.

Responsibilities

  • Contribute to infrastructure that supports distributed training workflows (e.g., PyTorch, Ray)
  • Work with workflow orchestration tools (e.g., Airflow, Flyte, or similar) to support multi-stage ML pipelines
  • Partner with ML engineers to support experimentation and model iteration
  • Help optimize performance and efficiency across data processing and training systems
  • You'll help build and evolve the infrastructure that powers training data generation, ML workflows, and distributed model training.
  • Relocation support is not available for this position
  • Our differences are strengths that enable us to support the growing and evolving needs of our customers, partners, and collaborators.

Requirements

  • Experience with Python and working with data-intensive workloads
  • Familiarity with ML frameworks (e.g., PyTorch, TensorFlow) and/or distributed systems (e.g., Ray, 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
  • Experience with workflow orchestration systems (Airflow, Flyte, etc.)

Skills

  • Improve reproducibility and reliability through dataset validation, monitoring, and testing
  • Contribute to the evolution of our offline ML platform architecture as it scales

Compensation

  • Note: This range reflects the anticipated base salary for this position.

Benefits

  • We offer a wide range of benefits designed to support well-being and work-life balance.
  • Please note: Benefits eligibility, specific offerings, and coverage vary based on the country and employment status.
  • Unity also enables teams across industries like automotive, manufacturing, and healthcare to design, simulate, and collaborate in 3D - closing the gap between ideas and reality.
  • If you have a disability that means there are preparations or accommodations we can make to help ensure you have a comfortable and positive interview experience, please fill out this form to let us know.
  • Beyond base salary, this role may be eligible for equity awards and participation in our company incentive plans (such as annual discretionary bonuses or sales commissions).
  • This is an ideal role for a recent graduate who is excited to work on large-scale systems and apply research-driven thinking to real-world machine learning problems.

Company info

  • Unity Vector builds an offline ML platform that powers insight, experimentation, attribution, and AI-driven decision-making across the company.
  • At Unity, we want our team members to thrive.

Equal opportunity

  • equal opportunity employer.

Visa & Work Authorization

  • Work visa/immigration sponsorship is not available for this position
Apply directly at Unity3d →Create a free account for alerts like thisView Unity3d immigration profile

This listing is sourced directly from Unity3d's careers page and normalized into a canonical job model.