Research Engineer, Evaluation

12 Hours ago • All levels • Research & Development

About the job

Job Description

Captions seeks a Research Engineer to push the boundaries of large-scale multimodal video diffusion models. Responsibilities include designing novel architectures, developing training techniques, and scaling models to billions of parameters. The role involves researching generative modeling, multimodal fusion, temporal modeling, video control, and temporal video editing. The engineer will train and optimize models at massive scale, developing distributed training approaches and implementing model surgery techniques. This involves creating new loss functions, conducting empirical studies, and validating research through product deployment and user feedback. The ideal candidate possesses a Master's or PhD in a related field, strong deep learning expertise (PyTorch, CUDA), and experience with large-scale model training and optimization.
Must have:
  • PhD/Master's in CS/ML
  • Experience with large-scale models
  • Deep learning expertise (PyTorch)
  • Generative modeling experience
  • Strong research background
Good to have:
  • Experience with 200B+ parameter models
  • FSDP, DeepSpeed, Megatron-LM experience
Perks:
  • Comprehensive medical, dental, and vision
  • 401k with employer match
  • Catered lunch
  • Dinner stipend
  • Doordash DashPass
  • Health & Wellness Perks
  • Team offsites
  • Generous PTO

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) **

Overview

Captions is seeking an exceptional Research Engineer to advance the state-of-the-art in large-scale multimodal video diffusion models. You'll conduct novel research on generative modeling architectures, develop new training techniques, and scale models to billions of parameters. As a key member of our ML Research team, you'll work at the cutting edge of multimodal generation while building systems that enable natural, controllable video creation. We're already training large-scale models with demonstrated product impact, and we're excited to continue expanding the scope and capabilities of our research.

We're especially excited about pushing the boundaries of audio-video generation, with a focus on realistic and charismatic human behavior that enables natural storytelling and creative iteration. Our models power creative tools used by millions of creators, and we're tackling fundamental challenges in how to generate compelling human motion, expression, and speech. 

Key Responsibilities

Research & Architecture Development:

  • Design and implement novel architectures for large-scale video and multimodal diffusion models

  • Develop new approaches to multimodal fusion, temporal modeling, and video control

  • Research temporal video editing techniques and controllable generation

  • Research and validate scaling laws for video generation models

  • Create new loss functions and training objectives for improved generation quality

  • Drive rapid experimentation with model architectures and training strategies

  • Validate research directly through product deployment and user feedback

Model Training & Optimization:

  • Train and optimize models at massive scale (10s-100s of billions of parameters)

  • Develop sophisticated distributed training approaches using FSDP, DeepSpeed, Megatron-LM

  • Design and implement model surgery techniques (pruning, distillation, quantization)

  • Create new approaches to memory optimization and training efficiency

  • Research techniques for improving training stability at scale

  • Conduct systematic empirical studies of architecture and optimization choices

Technical Innovation:

  • Advance state-of-the-art in video model architecture design and optimization 

  • Develop new approaches to temporal modeling for video generation

  • Create novel solutions for multimodal learning and cross-modal alignment

  • Research and implement new optimization techniques for generative modeling and sampling

  • Design and validate new evaluation metrics for generation quality

  • Systematically analyze and improve model behavior across different regimes

Preferred Qualifications

Research Experience:

  • Master's or PhD in Computer Science, Machine Learning, or related field

  • Track record of research contributions at top ML conferences (NeurIPS, ICML, ICLR)

  • Demonstrated experience implementing and improving upon state-of-the-art architectures

  • Deep expertise in generative modeling approaches (diffusion, autoregressive, VAEs, etc.)

  • Strong background in optimization techniques and loss function design

  • Experience with empirical scaling studies and systematic architecture research

Technical Expertise:

  • Strong proficiency in modern deep learning tooling (PyTorch, CUDA, Triton, FSDP, etc.)

  • Experience training diffusion models with 10B+ parameters

  • Experience with very large language models (200B+ parameters) is a plus

  • Deep understanding of attention, transformers, and modern multimodal architectures

  • Expertise in distributed training systems and model parallelism

  • Proven ability to implement and improve complex model architectures

  • Track record of systematic empirical research and rigorous evaluation

Engineering Capabilities:

  • Ability to write clean, modular research code that scales

  • Strong software engineering practices including testing and code review

  • Experience with rapid prototyping and experimental design

  • Strong analytical skills for debugging model behavior and training dynamics

  • Facility with profiling and optimization tools

  • Track record of bringing research ideas to production

  • Experience maintaining high code quality in a research environment

Team Culture

You'll work directly alongside our research and engineering teams in our NYC office. We've intentionally built a culture where technical innovation and research excellence are highly valued - your success will be measured by your contributions to improving our models and advancing the field, not by your ability to navigate politics. We're a team that loves diving deep into complex technical problems and emerging with practical breakthroughs.

  • 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

  • What sets us apart:

    • Opportunity to advance the state-of-the-art in video generation

    • Direct impact on products used by millions of creators

    • Access to massive compute resources and diverse, large-scale datasets

    • Environment that values both research excellence and practical impact

    • Ability to validate research through direct product feedback

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

View Full Job Description

Add your resume

80%

Upload your resume, increase your shortlisting chances by 80%

About The Company

New York, United States (On-Site)

New York, New York, United States (On-Site)

New York, New York, United States (On-Site)

New York, New York, United States (On-Site)

New York, New York, United States (On-Site)

New York, New York, United States (On-Site)

New York, New York, United States (On-Site)

New York, New York, United States (On-Site)

New York, New York, United States (On-Site)

New York, New York, United States (On-Site)

View All Jobs

Get notified when new jobs are added by Captions

Similar Jobs

Luxoft - C Developer

Luxoft, India (On-Site)

Epic Games - Research Programmer

Epic Games, Canada (On-Site)

Yext - UI Designer

Yext, India (On-Site)

Fabric - Applied Cryptographer, ZKP Research

Fabric, United States (Remote)

Scanline VFX - Pipeline Supervisor

Scanline VFX, India (Hybrid)

Rocket - Database/Operating Systems Internals Engineer

Rocket, United States (Remote)

Riot Games - VFX Artist II - Gameplay, Teamfight Tactics

Riot Games, United States (On-Site)

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Futuristic Labs - Prototype Engineer

Futuristic Labs, India (On-Site)

PlayStation Global - Senior People Tech & Services Analyst-Talent Systems

PlayStation Global, United States (On-Site)

Attentive - Staff Software Engineer

Attentive, United States (Remote)

Nintendo - Intern – Networking Software Engineer (NTD)

Nintendo, United States (On-Site)

DraftKings - Product Designer, Platform

DraftKings, United States (Remote)

Luminar Technologies - Product Security Lead

Luminar Technologies, India (On-Site)

CloudHire - UI UX Developer

CloudHire, India (Remote)

Enphase Energy - Staff UI/UX Designer

Enphase Energy, India (On-Site)

Scopely - Lead UI Artist- Monopoly GO!

Scopely, Spain (Hybrid)

Get notifed when new similar jobs are uploaded

Jobs in New York, New York, United States

Warner Bros Discovery - Writer, Streaming (Temporary)

Warner Bros Discovery, United States (On-Site)

Fluence - Director of Planning, Americas

Fluence, United States (Hybrid)

Crunchyroll - Software Engineer III, Payments

Crunchyroll, United States (Remote)

Blizzard Entertainment - Senior II, 3D Artist - Hearthstone | Irvine, CA

Blizzard Entertainment, United States (Hybrid)

Nintendo - Senior UX Writer

Nintendo, United States (Hybrid)

Nagarro - Associate Engineer, Tech support

Nagarro, United States (On-Site)

Nintendo - Lead Audio Algorithms Software Engineer (NTD)

Nintendo, United States (On-Site)

The Walt Disney Company - Associate Producer - WPVI

The Walt Disney Company, United States (On-Site)

Get notifed when new similar jobs are uploaded

Research & Development Jobs

Axon - Senior Machine Learning Engineer

Axon, United States (On-Site)

Microsoft - Principal Software Engineer - Camera Drivers

Microsoft, United States (On-Site)

Nintendo - Senior Engineer, CPU Debugger (NTD)

Nintendo, United States (On-Site)

Microsoft - Research Intern - Office AI

Microsoft, United States (On-Site)

Google - Silicon Debug Engineer

Google, India (On-Site)

Microsoft - Research Intern - Knowledge and Reasoning

Microsoft, United States (On-Site)

Microsoft - Principal IP/SOC Design Engineer

Microsoft, India (On-Site)

Get notifed when new similar jobs are uploaded