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
Average salary estimate
If an employer mentions a salary or salary range on their job, we display it as an "Employer Estimate". If a job has no salary data, Rise displays an estimate if available.
Analyst for TKO Global Analytics & Insights to build dashboards, run analyses, and translate data into strategic insights across TKO properties.
Mount Carmel seeks a part-time Nuclear Medicine Technologist to perform diagnostic nuclear medicine studies, maintain radiation safety, and deliver patient-focused imaging services at its Westerville-area facility.
Experienced backend engineer needed to lead development of scalable APIs and services for a cutting-edge investor identity platform in a remote-first company.
Help design and implement Zoox's robotics middleware—building real-time, cross-platform C++ systems that enable safe, scalable autonomous vehicle software.
Experienced AI Developer needed to build and enhance LLM-powered tooling and full-stack solutions for digital modeling at a government-focused consulting firm.
Experienced Salesforce developer needed to lead Salesforce application development and integrations (Apex, Lightning, MuleSoft) in support of VBA SaaS/PaaS solutions for a federal SDVOSB.
Pinterest is hiring a Manager II, Production Engineering to lead and grow a team that drives reliability, scalability and performance improvements across large-scale platform infrastructure.
Experienced platform-focused engineer needed to design and maintain internal developer tools and infrastructure that boost engineering productivity across a remote-friendly US-based organization.
Booz Allen is hiring a senior Full Stack Software Developer to build cloud-native, end-to-end solutions (AWS, IaC, PWA) for cleared government programs requiring TS/SCI with polygraph.
Lead LinkedIn's Jobs Marketplace Core team as an Engineering Manager to drive AI-powered marketplace optimization, scale distributed systems, and mentor a team delivering value for job seekers and hirers.
Booz Allen is hiring an AI-focused Software Engineer in Arlington, VA to build and deploy production-grade, secure AI/ML systems and integrations for mission-driven clients.
Topline Pro is hiring a Full Stack Engineer to build scalable web applications across Python/Django backends and React/TypeScript frontends to accelerate growth for service businesses.
Lead and grow a behavior engineering team at Applied Intuition to architect and deliver L4 trucking behavior systems for high-traffic highway and suburban environments.
Experienced engineers are sought to design and implement web applications and microservices using React and Golang in a remote-first, collaborative environment with growth and mentoring opportunities.
Workiva is hiring a Senior Software Engineer to lead the design and delivery of scalable, production-grade data pipelines and streaming workflows for analytics and business-critical systems.
SpringRole is the first professional reputation network powered by artificial intelligence and blockchain to eliminate fraud from user profiles. Because SpringRole is built on blockchain and uses smart contracts, it's able to verify work experienc...
397 jobs
🔮
Hi, I'm Risa! Your AI Career Copilot
you, just ask me below!
