Software Engineer, ML Infra
About Unit
Unit is an AI video company. We’re building GTV, a consumer product that combines creativity and automation to deliver a next-generation video experience. Our team includes senior people from Instagram, TikTok, and NVIDIA. We’re based in San Francisco and backed by Khosla Ventures, Air Street Capital, and other leaders in AI.
Role Overview
We're looking for an experienced ML Infra Software Engineer to join the GTV team. You'll work closely with researchers and engineers to build our core ML systems, including training and serving generative models, orchestrating real-time data pipelines, enabling reinforcement learning at scale, and more. This role is for someone who loves building high-performance ML infra and wants to reimagine the future of media.
Responsibilities
Design and develop scalable ML systems that power the creative engine behind GTV
Partner with researchers and engineers to turn new model/product ideas into production-ready workflows
Build robust pipelines for real-time data collection, labeling, and feedback
Drive reliability and efficiency of distributed systems with automated testing, performance benchmarking, and end-to-end observability
Requirements
5+ years of software engineering experience and proficiency in Python
Industry experience designing and operating large-scale distributed systems for ML training and serving
Deep understanding of multi-node training, reinforcement learning, large-scale data processing, and real-time feedback loops
Proven ability to debug complex systems, navigate ambiguity, and take ownership of challenging problems
Excellent communication and collaboration with cross-functional teams (research, engineering, and design)
Nice To Have
Prior work experience at early-stage or high-growth startups
Familiarity with video generation, evaluation, and labeling
Background in generative AI and infra for consumer-facing products
Compensation
$225-$275 base salary
Competitive equity grant
Top-tier benefits (medical, dental, and vision)
Daily dinner stipend
Relocation support