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

Information Technology

5K-10K (6748)

Founded 2004

123 open positions

Links

View All Jobs

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

1 month 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 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 123

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.
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.
English proficiency for frequent professional verbal and written communication with global colleagues and partners.

Benefits

Gross pay salary of $112,700 to $169,100 USD.
Comprehensive health, life, and disability insurance.
Competitive retirement/pension plans and employee stock ownership.
Generous vacation and personal days.
Support for new parents through leave and family-care programs.
Mental health and wellbeing programs and support.
Training and development programs.
Commute subsidy and office food snacks.

Interested in this position?

Apply directly on the company website

Apply Now

Similar Roles

Software Engineer II, Backend (ML Training & Serving)

Affirm 1K-5K Diversified Financial Services

Affirm is hiring a Software Engineer II for its ML Training & Serving engineering team to build the infrastructure that trains and serves machine learning models across the company.

Canada Full-time Junior Backend Engineer Machine Learning Engineer

$89k-$126k

AWS Kotlin Kubernetes Machine Learning MySQL Python

9 hours, 45 minutes ago

Apply

9 hours, 45 minutes ago

Ssr. Fullstack Engineer

Resilient Co 11-50 Professional Services

Resilient Co. is hiring a semi-senior Fullstack Engineer in Argentina or Brazil to build AI-driven full-stack solutions for enterprise workflows, with a focus on agentic AI, machine learning, backend services, and cloud integration.

Argentina Brazil Contract Senior Full-stack Engineer Machine Learning Engineer

Angular Azure C# CI/CD Django Docker Entity Framework FastAPI Flask Git JavaScript Microservices .NET NumPy Pandas Python RabbitMQ React Scikit-learn Terraform Vue.js YAML

10 hours ago

Apply

10 hours ago

[Job 29881] Senior Machine Learning Engineer, Brazil

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

CI&T is hiring a Senior Machine Learning Engineer in Brazil to develop and deploy production ML solutions that turn data and AI capabilities into measurable business impact.