Cantina Labs · 21 hours ago
Member of Technical Staff, Data & ML Infrastructure for Video Models
United States
Full-time
Remote
Mid, Senior Level
$200K/yr - $260K/yr
3+ years expCantina Labs is a social AI company that develops advanced real-time models to enhance storytelling and creativity. They are seeking a Member of Technical Staff to build and scale data pipelines for large video generation models, focusing on data collection, preparation, and workflow development.
Artificial Intelligence (AI)Social MediaSoftwareSocial Network
Responsibilities
Build and maintain data pipelines for large video generation models, including data ingestion, parsing, filtering, preprocessing, and dataset curation at scale, using tools such as AWS S3 and DynamoDB
Design and run annotation workflows across platforms such as MTurk, Prolific, and Mechanical Turk, including task design, quality control, and label validation
Train, evaluate, and improve smaller supporting models used for data filtering, quality assessment, preprocessing, or other parts of the ML pipeline
Partner closely with research and engineering teams to turn experimental workflows into scalable, repeatable systems that support model training and evaluation
Own data quality across the pipeline by identifying bottlenecks, failure modes, and low-quality sources, and continuously improving tooling and processes
Build internal tools and automation that make it easier to prepare datasets, launch annotation jobs, monitor outputs, and support model development end to end
Drive larger pipeline projects from start to finish, such as new dataset creation efforts or upgrades to labeling and preprocessing infrastructure
Work within a Kubernetes-based training infrastructure, ensuring datasets are properly prepared, formatted, and delivered to training clusters
Profile and optimize research model inference scripts used in preprocessing steps, ensuring that model-driven filtering and transformation stages run within practical time and cost constraints when applied to large-scale raw data
Qualification
Required
3+ years of experience in machine learning, applied ML, data pipelines, or related engineering roles, ideally working on large-scale multimodal, video, or vision-based systems
Strong programming skills in Python and solid experience building reliable data processing and preprocessing pipelines for ML workflows
Hands-on experience preparing training data for ML models, including parsing, filtering, dataset curation, quality control, and large-scale data handling using tools such as AWS S3 and DynamoDB
Familiarity with annotation and labeling workflows, including task design, vendor or crowd-platform orchestration such as MTurk or Prolific, and methods for ensuring label quality
Experience working with Kubernetes for orchestrating distributed workloads, including data preprocessing, pipeline execution, and dataset delivery to training clusters
Comfort working across cloud and on-demand compute environments such as AWS and RunPod, with the ability to port and optimize pipelines across infrastructure
Familiarity with distributed data processing frameworks and experience designing systems that operate reliably at scale across many nodes or workers
Working knowledge of PyTorch and the broader deep learning stack, with the ability to read, debug, and optimize research model inference code for use in production preprocessing pipelines
Ability to work cross-functionally with research and engineering teams and translate experimental ideas into robust, scalable systems
Bachelor's, Master's, or PhD in Computer Science, Machine Learning, Engineering, Mathematics, or a related technical field
Preferred
Experience training, evaluating, or fine-tuning smaller ML models used for classification, filtering, ranking, quality assessment, or other supporting tasks in an ML pipeline
Experience in generative video, computer vision, or multimodal ML is strongly preferred
Benefits
Competitive salary and generous company equity
Medical, dental, and vision insurance 6.99% of premiums covered by Cantina
42 days of paid time off, including:
+ 15 PTO days
+ 10 sick days
+ 15 company holidays
+ 2 floating holidays
Generous parental leave & fertility support
401(k) retirement savings plan
Lifestyle spending account 6500/month to use however youd like
Complimentary lunch and snacks for in-office employees
One Medical membership, and more!
Company
Cantina Labs
Cantina Labs is a social AI company, developing a suite of advanced real-time models that push the boundaries of expression, personality, and realism.
51-200 employees
H1B Sponsorship
Cantina Labs has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
64%
Represents job field similar to this job
Engineering and Development
Product Management
Customer Service and Support
Creatives and Design
Trends of Total Sponsorships
*2026 (1)
2025 (9)
2024 (3)
2023 (1)
Funding
Current Stage
Growth StageTotal Funding
unknown2025-01-01Seed
Leadership Team
Daniel Klaus
President & Co-Founder
Prakash Ramakrishna
Co-Founder
Recent News
2026-05-27
News Archives - Times Kuwait
2026-02-08
Company data provided by crunchbase