Member of Technical Staff, MLOps

Job Description

Job Description

Captions is the leading video AI company, building the future of video creation. Over 10 million creators and businesses have used Captions to create videos for social media, marketing, sales, and more. We're on a mission to serve the next billion.

We are a rapidly growing team of ambitious, experienced, and devoted engineers, researchers, designers, marketers, and operators based in NYC. You'll join an early team and have an outsized impact on the product and the company's culture.

We’re very fortunate to have some the best investors and entrepreneurs backing us, including Index Ventures (Series C lead), Kleiner Perkins (Series B lead), Sequoia Capital (Series A and Seed co-lead), Andreessen Horowitz (Series A and Seed co-lead), Uncommon Projects, Kevin Systrom, Mike Krieger, Lenny Rachitsky, Antoine Martin, Julie Zhuo, Ben Rubin, Jaren Glover, SVAngel, 20VC, Ludlow Ventures, Chapter One, and more.

Check out our latest financing milestone and some other coverage :

The Information : 50 Most Promising Startups

Fast Company : Next Big Things in Tech

The New York Times : When A.I. Bridged a Language Gap, They Fell in Love

Business Insider : 34 most promising AI startups

Time : The Best Inventions of 2024

• Please note that all of our roles will require you to be in-person at our NYC HQ (located in Union Square)

About the Role :

Captions seeks an exceptional MLOps Research Engineer (MOTS) to architect and scale the machine learning infrastructure for our rapidly growing creative platform used by millions. You'll own the development of our distributed training systems, optimize our rapidly growing GPU clusters, and build performant inference pipelines that power our cutting-edge multimodal video diffusion models. As a key member of our ML Research team in a fast-growing Series C startup, you'll create foundational infrastructure enabling rapid research iteration while maintaining production-grade reliability and efficiency. We're already training large-scale models and are excited to dramatically expand our infrastructure capabilities.

Key Responsibilities :

Core Systems Development :

Develop and optimize distributed training frameworks integrating multiple modalities (video, audio, text, and structured metadata)

Build flexible systems for cross-modal training orchestration and efficient experimentation

Design reproducible training environments with versioned dependencies and configurations

Implement comprehensive testing frameworks for validating model training correctness and performance

Create infrastructure for systematic model quality assessment and performance benchmarking

Infrastructure Development :

Design and implement flexible training orchestration systems that balance research agility with large-scale model training

Build robust monitoring and observability systems for complex training and inference pipelines

Design and manage GPU clusters optimized for distributed training of multimodal models

Build out comprehensive automated metrics collection and alerting across our ML stack

System Optimization :

Profile and optimize model training throughput using mixed precision, gradient checkpointing, and advanced memory techniques

Develop custom CUDA and Triton kernels to accelerate critical compute paths

Implement creative solutions for cost optimization across spot instances and reserved capacity

Design and optimize real-time inference systems enabling fast research iteration cycles

Research & Product Impact :

Build infrastructure enabling rapid testing of research hypotheses

Create systems supporting close collaboration between infrastructure and research teams

Develop frameworks for reproducible research experimentation

Enable seamless deployment of research innovations to production

Requirements :

Technical Background :

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

Strong programming skills in Python and systems programming

Experience with distributed systems and scalable infrastructure

Track record of building reliable, performant large-scale ML systems

Areas of Expertise (Strong experience in some or all of these areas) :

Deep expertise in PyTorch internals and distributed training frameworks (FSDP, DeepSpeed)

GPU cluster management and optimization

Performance profiling and systems optimization

CUDA programming and kernel optimization

Containerization and orchestration (Docker, Kubernetes)

ML model serving and deployment at scale

Language models and attention mechanism optimization

Video and audio processing pipelines

Large-scale diffusion models

Engineering Approach :

Love diving deep into complex systems optimization challenges

Take ownership of critical infrastructure while collaborating effectively

Get excited about pushing the boundaries of ML system performance

Want to work directly with researchers on cutting-edge ML problems

Thrive in fast-paced, research-driven environments

About the Team :

You'll work full-time, on-site in our NYC office alongside researchers and engineers who are dedicated to building world-class generative models and data infrastructures. We've intentionally built a culture that prizes open discussion of technical approaches, rapid iteration, and direct access to decision makers. Your success will be measured by the performance and reliability of our systems, enabling our researchers to iterate quickly on and develop ambitious ideas. You'll have significant autonomy to shape our infrastructure direction and direct impact on our ability to serve millions of creators.

Our team values :

Open technical discussions and collaboration

Rapid iteration and practical solutions

Deep technical expertise and continuous learning

Direct impact on research and product outcomes

Benefits :

Comprehensive medical, dental, and vision plans

401K with employer match

Commuter Benefits

Catered lunch multiple days per week

Dinner stipend every night if you're working late and want a bite!

Doordash DashPass subscription

Health & Wellness Perks (Talkspace, Kindbody, One Medical subscription, HealthAdvocate, Teladoc)

Multiple team offsites per year with team events every month

Generous PTO policy and flexible WFH days

Captions provides equal employment opportunities to all employees and applicants for employment and prohibits discrimination and harassment of any type without regard to race, color, religion, age, sex, national origin, disability status, genetics, protected veteran status, sexual orientation, gender identity or expression, or any other characteristic protected by federal, state or local laws.

Please note benefits apply to full time employees only.

Compensation Range : $160K - $250K

Salary : $160,000 - $250,000