Impress employers and recruiters.
Choose from hundreds of resume examples.

Target Resume Now

About the Role

Unity Vector builds an offline ML platform that powers insight, experimentation, attribution, and AI-driven decision-making. We are seeking a Machine Learning Engineer to join our Offline Infrastructure team, ideal for a recent graduate interested in large-scale systems and applying research to real-world ML problems. You will help build and evolve the infrastructure for training data generation, ML workflows, and distributed model training, working closely with experienced engineers and researchers to ensure ML pipelines are reliable, scalable, and efficient.

Responsibilities

Build and maintain data pipelines for generating training datasets and experimentation.
Contribute to infrastructure supporting distributed training workflows (e.g., PyTorch, Ray).
Work with workflow orchestration tools (e.g., Airflow, Flyte) for multi-stage ML pipelines.
Improve reproducibility and reliability through dataset validation, monitoring, and testing.
Partner with ML engineers to support experimentation and model iteration.
Optimize performance and efficiency of data processing and training systems.
Contribute to the evolution of the offline ML platform architecture.

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 (through research or projects).
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.
Interest in building scalable, reliable infrastructure for machine learning.
Sufficient knowledge of English for professional verbal and written exchanges.

Skills

Python
PyTorch
TensorFlow
Ray
Spark
Airflow
Flyte

Location

Mountain View
SF/Bay
Bellevue
Seattle
NYC
Remote CA

Work Type

Remote

Experience Level

Recent university graduate

Education Level

Bachelor's degree in Computer Science, Machine Learning, Systems, or a related field

Salary/Compensations

Mountain View, SF/Bay: $117,000 - $152,000 gross USD
Bellevue, Seattle, NYC, Remote CA: $104,100 - $135,300 gross USD

Benefits

Comprehensive health, life, and disability insurance
Commute 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

About the Company

Unity [NYSE: U] is the world’s leading game engine, powering play for more than 3 billion consumers each month. The top mobile games in the world, the most played PC indie titles, the most innovative console games, and virtually all of the top XR and Web Games are developed, deployed, and grown in Unity. 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. For more information, please visit www.unity.com.

Equal Opportunity

Unity is a proud equal opportunity employer. We are committed to fostering an inclusive, innovative environment and celebrate our employees across age, race, color, ancestry, national origin, religion, disability, sex, gender identity or expression, sexual orientation, or any other protected status in accordance with applicable law. Our differences are strengths that enable us to support the growing and evolving needs of our customers, partners, and collaborators. 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.
Pursuant to the San Francisco Fair Chance Ordinance, we will consider for employment qualified applicants with arrest and conviction records.