Machine Learning Engineering, Model Training

3 Months ago • 3 Years + • Data Analyst • Artificial Intelligence

Job Summary

Job Description

Captions seeks a skilled Machine Learning Engineer to build data infrastructure for training cutting-edge video generation models. Responsibilities include designing data pipelines for efficient video data handling, optimizing video pre-processing, creating data loaders for large-scale datasets, implementing feature engineering techniques, collaborating with research and engineering teams to scale infrastructure, and managing cluster code for high-performance training. This role involves building foundational, state-of-the-art machine systems from scratch and requires strong Python programming skills and experience with large-scale data processing frameworks and deep learning systems.
Must have:
  • Python programming skills
  • Data pre-processing & feature engineering
  • Large-scale data processing frameworks
  • Deep learning systems
  • Offline model training workflows
  • Data loaders & cluster infrastructure
Perks:
  • Comprehensive medical, dental, and vision plans
  • 401K with employer match
  • Commuter Benefits
  • Catered lunch
  • Dinner stipend
  • Doordash DashPass subscription
  • Health & Wellness Perks
  • Team offsites & events
  • Generous PTO and flexible WFH days

Job Details

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:

We’re seeking a skilled Machine Learning Engineer to join our AI Research team and build the data infrastructure that powers the training of cutting-edge video generation models.
In this role, you’ll develop offline jobs dedicated to training large generative models, manage training cluster code, and create data loaders to handle large-scale video datasets. Being an early member of our AI Research team will give you the opportunity to build foundational, state-of-the-art machine systems 0 to 1. 

Key Responsibilities:

  • Design and develop robust data pipelines to support the efficient handling and processing of video data, ensuring high-quality data input for model training.

  • Build and optimize systems for video frame extraction and other pre-processing steps to prepare data for training workflows.

  • Create and manage data loaders for large-scale video datasets, focusing on speed and efficiency to support various model training requirements.

  • Implement feature engineering techniques that enhance data quality and diversity, aiding in model accuracy and performance.

  • Collaborate with research and engineering teams to scale data infrastructure and enable seamless experimentation and model iterations.

  • Write and maintain cluster code to support high-performance training operations, including resource allocation and management.

Requirements:

  • Bachelor’s or Master’s degree in Computer Science, Data Engineering, Machine Learning, or a related field.

  • 3+ years of professional experience in software engineering, data engineering, or ML infrastructure development.

  • Strong programming skills, particularly in Python, with proven experience with data pre-processing and feature engineering, ideally within video or image data contexts.

  • Professional experience working with large-scale data processing frameworks, deep-learning systems, offline model training workflows, data loaders, and cluster infrastructure.

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: $170K - $250K

Similar Jobs

Meta - Software Engineering Manager, Product

Meta

Sunnyvale, California, United States (On-Site)
3 Months ago
Forcepoint - Software Engineer - I-1 (Java/C++/Python, linux )

Forcepoint

Mumbai, Maharashtra, India (On-Site)
4 Months ago
SatSure - Senior Machine Learning Researcher

SatSure

Bengaluru, Karnataka, India (On-Site)
5 Months ago
Samsung Semiconductor - Principal Engineer, NPU architect

Samsung Semiconductor

San Jose, California, United States (Hybrid)
4 Months ago
Palo Alto Networks - Domain Consultant - Security Operations Transformation

Palo Alto Networks

Newark, New Jersey, United States (Remote)
3 Months ago
Highspot - Sr. Data Analytics Manager - Remote

Highspot

United States (Remote)
4 Months ago
Visa - Data Science Intern 2025

Visa

Thai Binh, Vietnam (On-Site)
4 Months ago
Lifechurch - Senior Data Engineer

Lifechurch

Edmond, Oklahoma, United States (On-Site)
4 Months ago
Hoyoverse - Legal Counsel (Privacy and Data Protection)

Hoyoverse

Singapore (On-Site)
7 Months ago
paypal - MTS 1 Data Engineer

paypal

Shanghai, Shanghai, China (On-Site)
4 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Marvell - Principal Engineer AI/ML

Marvell

Bengaluru, Karnataka, India (On-Site)
4 Months ago
Social Discovery Group - ML Ops (The.One)

Social Discovery Group

Serbia (Remote)
4 Months ago
Netflix - Data Engineer (L5)

Netflix

United States (Remote)
3 Months ago
Glean - Data Scientist, Growth

Glean

Palo Alto, California, United States (On-Site)
3 Months ago
Google - Software Engineer III, Infrastructure, Google Cloud, Developer X

Google

Warsaw, Masovian Voivodeship, Poland (On-Site)
3 Months ago
ByteDance - Student Researcher (Doubao (Seed) - Foundation Model - Speech Understanding) - 2025 Start (PhD)

ByteDance

San Jose, California, United States (On-Site)
3 Months ago
Rockstar Games - Senior Full Stack Engineer (C#/React)

Rockstar Games

New York, New York, United States (On-Site)
5 Months ago
ION - Infrastructure Engineer, Budapest

ION

Budapest, Hungary (Hybrid)
4 Months ago
ness - Big Data Engineers (T5&6)

ness

Romania (On-Site)
3 Months ago
Picarro - DevOps Manager

Picarro

Bengaluru, Karnataka, India (On-Site)
4 Months ago

Get notifed when new similar jobs are uploaded

Jobs in New York, New York, United States

Zoox - Senior/Staff Software Engineer - Simulation Traffic & Behavior Modeling

Zoox

Foster City, California, United States (Hybrid)
4 Months ago
eBay - Senior Counsel, Payments

eBay

San Jose, California, United States (Hybrid)
4 Months ago
Patreon - Scientist to Machine Learning Engineer

Patreon

New York, New York, United States (Hybrid)
3 Months ago
Overwolf - Brand Partnerships Director Central

Overwolf

Chicago, Illinois, United States (On-Site)
4 Months ago
ByteDance - Senior Data Engineer, Global Payments

ByteDance

San Jose, California, United States (On-Site)
3 Months ago
ByteDance - Business Analyst (Multiple Positions)

ByteDance

Seattle, Washington, United States (On-Site)
3 Months ago
Super.com - Senior Full-Stack Software Engineer ( Remote! )

Super.com

Boston, Massachusetts, United States (Remote)
3 Months ago
paypal - Audit Associate, Internal Audit

paypal

Chicago, Illinois, United States (Hybrid)
3 Months ago
Blizzard Entertainment - UX Designer - World of Warcraft | Irvine, CA

Blizzard Entertainment

Irvine, California, United States (Hybrid)
3 Months ago
Salesforce - Enterprise Account Executive, Tableau-State of California

Salesforce

California, United States (On-Site)
4 Months ago

Get notifed when new similar jobs are uploaded

Data Analyst Jobs

Playrix - Data QA Engineer

Playrix

Serbia (Remote)
4 Months ago
Playrix - Lead Business Analyst (DWH)

Playrix

Cyprus (Remote)
4 Months ago
Playkot - SENIOR PRODUCT ANALYST

Playkot

Belgrade, Serbia (On-Site)
3 Months ago
PwC - IFS-Operate-Data Analytics- Manager 2-Kolkata

PwC

Kolkata, West Bengal, India (On-Site)
4 Months ago
Meta - GRC Analyst

Meta

Menlo Park, California, United States (On-Site)
3 Months ago
The Walt Disney Company - Statistics Analyst II

The Walt Disney Company

Bristol, Connecticut, United States (On-Site)
3 Months ago
Dew Software - Data Scientist

Dew Software

Newark, New Jersey, United States (On-Site)
3 Months ago
PwC - IN-Senior Associate_Big Data Engineer_Data & Analytics_Advisory_ PAN India

PwC

Gurugram, Haryana, India (On-Site)
4 Months ago
Google - Customer Engineer I, Data Analytics, Google Cloud

Google

San Francisco, California, United States (On-Site)
3 Months ago

Get notifed when new similar jobs are uploaded

About The Company

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)

San Francisco, California, United States (On-Site)

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

View All Jobs

Get notified when new jobs are added by Captions

Level Up Your Career in Game Development!

Transform Your Passion into Profession with Our Comprehensive Courses for Aspiring Game Developers.

Job Common Plug