Machine Learning Engineer, Offline Infrastructure

Role Overview

Unity Vector builds an offline ML platform that powers insight, experimentation, attribution, and AI-driven decision-making across the company. We’re looking for a Machine Learning Engineer to join our Offline Infrastructure team.

What You Will Do

Build and maintain data pipelines, contribute to infrastructure that supports distributed training workflows, work with workflow orchestration tools, improve reproducibility and reliability, partner with ML engineers, and help optimize performance and efficiency.

Why It Might Be a Fit

This role offers the opportunity to bridge research and production—translating advanced ideas into systems that operate at scale. You’ll work closely with experienced engineers and researchers, and contribute to systems that ensure our ML pipelines are reliable, scalable, and efficient.

Requirements

• Bachelor's Degree in Computer Science, Machine Learning, Systems, or a related field
• Strong foundation in machine learning systems, distributed systems, or large-scale data processing
• 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 with data pipelines, model training workflows, or large datasets

Benefits

• Comprehensive health, life, and disability insurance
• Commutute subsidy
• Employee stock ownership
• Competitive retirement/pension plans
• Generous vacation and personal days
• Support for new parents through leave and family-care programs
• Office food snacks
• Mental Health and Wellbeing programs and support
• Employee Resource Groups
• Global Employee Assistance Program
• Training and development programs
• Volunteering and donation matching program