ML Infrastructure Engineer - (Early Career/Internship)

About Unity Vector

Unity Vector develops an offline ML platform designed to drive insight, experimentation, attribution, and AI-driven decision-making across organizations. Our systems are built to handle massive batch and streaming data, supporting analytics, product intelligence, machine learning pipelines, and business operations. As data volumes and complexity increase, our platform becomes essential for large-scale model training, feature generation, and experimentation workflows that power production ML systems.

About the Role

We are seeking a Machine Learning Engineer to join our Offline Infrastructure team. This position is ideal for a recent graduate passionate about large-scale systems and applying research-driven thinking to real-world machine learning challenges. You will play a key role in building and enhancing the infrastructure responsible for training data generation, ML workflows, and distributed model training. Collaborating closely with experienced engineers and researchers, you will contribute to systems that ensure our ML pipelines are reliable, scalable, and efficient. This role offers a unique opportunity to translate advanced research into production-ready systems operating at scale.

What You'll Be Doing

• Build and maintain data pipelines for generating ML training datasets and supporting experimentation.
• Contribute to infrastructure that enables distributed training workflows (e.g., PyTorch, Ray).
• Work with workflow orchestration tools (e.g., Airflow, Flyte, or similar) to manage multi-stage ML pipelines.
• Enhance reproducibility and reliability through dataset validation, monitoring, and testing.
• Partner with ML engineers to support experimentation and model iteration.
• Optimize performance and efficiency across data processing and training systems.
• Contribute to the architectural evolution of our offline ML platform as it scales.

What We're Looking For

• Bachelor’s degree in Computer Science, Machine Learning, Systems, or a related field.
• A strong foundation in machine learning systems, distributed systems, or large-scale data processing (demonstrated through research or projects).
• Proficiency in Python and experience 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.
• Excellent problem-solving skills and the ability to translate research concepts into practical systems.
• A keen interest in building scalable, reliable infrastructure for machine learning.

Nice to Have

• Experience with workflow orchestration systems (Airflow, Flyte, etc.).
• Exposure to large-scale data platforms (data lakes, warehouses, streaming systems).
• Publications or research in ML systems, distributed systems, or related areas.

Additional Information

• Relocation support is not available for this position.
• Work visa/immigration sponsorship is not available for this position.

Benefits

At Unity, we are committed to supporting our team members' well-being and work-life balance. Benefits eligibility and offerings may vary by country and employment status. Our global benefits often include:

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

About Unity

Unity (NYSE: U) is the world's leading game engine, powering play for over 3 billion consumers monthly. Our technology is used in top mobile, PC, console, XR, and Web games, and enables industries like automotive, manufacturing, and healthcare to design, simulate, and collaborate in 3D. Learn more at www.unity.com.

*Unity is an equal opportunity employer committed to fostering an inclusive and innovative environment. We celebrate diversity and believe our differences are our strengths. If you require accommodations for a disability during the interview process, please fill out this form.

*This position requires professional English proficiency for effective verbal and written communication with global colleagues and partners.

*Headhunters and recruitment agencies should not submit unsolicited resumes. Unity will not pay fees for unsolicited submissions.

*Your privacy is important. Please review our Prospect Privacy Policy and Applicant Privacy Policy. Contact DPO@unity.com for privacy concerns.

*Note: The salary range reflects anticipated base pay. This role may also be eligible for equity awards and company incentive plans. The final offer will depend on factors like location, experience, background, and skills.

Salary Range: $112,700—$169,000 USD