Applied ML Engineer, Data
Employment Information
Report this job
Job expired or something wrong with this job?
Job Description
Applied ML Engineer at Cantina Labs building data pipelines for video generation models. Excited about AI's role in creativity and social interactions.
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.
Requirements:
- 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; experience in generative video, computer vision, or multimodal ML is strongly preferred.
- Bonus: Experience training, evaluating, or fine-tuning smaller ML models used for classification, filtering, ranking, quality assessment, or other supporting tasks in an ML pipeline.
Benefits:
- Competitive salary and generous company equity
- Medical, dental, and vision insurance – 99.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 – $500/month to use however you’d like
- Complimentary lunch and snacks for in-office employees
- One Medical membership, and more!
Similar Jobs
Director of AI & Machine Learning
Director of AI & Machine Learning leading AI capability across MicroTech to develop solutions and improve operational efficiency.
Machine Learning Scientist
Machine Learning Scientist developing generative models for biomolecular structure prediction at Iambic Therapeutics. Contributing to core ML technologies relevant for drug discovery.
Machine Learning Engineer
Machine Learning Engineer at Adapter focusing on deploying ML models and optimizing performance. Engaging with cross-functional teams to implement innovative AI solutions.
ML Engineer – Speech-to-Speech, Subject Matter Expert
Experienced ML Engineer advising on speech-to-speech pipeline for Vosyn, an AI startup. Collaborate to optimize TTS models and lead technical initiatives.
Senior Manager – Machine Learning
Senior Manager leading the Machine Learning team to enhance CCC’s subrogation products. Collaborating with stakeholders and delivering innovative AI solutions in the insurance technology space.
Senior AI/ML Engineer
Senior AI/ML Engineer designing and implementing workflows leveraging LLMs at iBusiness. Collaborating with cross-functional teams to develop scalable AI-driven solutions.
Machine Learning Engineer – Mid-level
Machine Learning Engineer responsible for developing and operating scalable ML solutions at Extractta. Collaborating with product, data, and engineering teams to deliver robust models.
Senior Machine Learning Engineer
Senior Machine Learning Engineer at IQVIA leading ML applications design focusing on generative AI solutions. Collaborating with teams and guiding development of production-grade algorithms.
Senior Machine Learning Engineer
Senior Machine Learning Engineer developing core ML models and data pipelines for a gaming company. Seeking an autonomous engineer with deep passion for video games.
Machine Learning Engineer
Machine Learning Engineer developing AI safety solutions at Gray Swan. Focus on real-world impact in adversarial testing and model evaluation.
Staff Machine Learning Engineer – Generative AI, Full-Stack Applications
Senior Machine Learning Engineer developing AI solutions and full-stack applications for CVS Health. Collaborating with teams to design and prototype secure, resilient products and platform components.
Senior Machine Learning Engineer – Consultant
Senior AI/ML Engineer Consultant responsible for implementing AI solutions and mentoring juniors at leading AI consultancy Provectus. Join to solve complex technical challenges and impact enterprise clients.
Machine Learning Engineer
Machine Learning Engineer designing and evaluating deep learning models for agricultural robotics applications. Collaborating with engineers and product teams to enhance model performance in real-world scenarios.
Senior Machine Learning Engineer – AI Product
Machine Learning Engineer for AI Product team at Qonto, developing customer-facing AI solutions. Collaborating with stakeholders and ensuring integration of ML solutions into financial services.
ML Engineer – Agents, Data
Senior ML Engineer for AI product development in a leading social discovery company. Working on large-scale ML systems and cutting-edge generative AI technologies.
Senior MLOps
MLOps role enhancing ML infrastructure and deployment for a global travel-tech company. Focus on managing ML services and optimizing resources in a fully remote setup.
MLOps Engineer
ML Ops Engineer designing and maintaining ML workflows using AWS services. Collaborating with data scientists to ensure reliable and scalable models in production.
Machine Learning Lead
Machine Learning Lead driving AI initiatives for LexisNexis Legal & Professional, delivering scalable, robust solutions with hands-on leadership and mentoring.
Senior ML Engineer
Senior AI Engineer contributing to AI engineering teams at Relativity. Design and develop scalable AI systems, collaborating with cross-functional teams and driving engineering excellence.
AI/ML Engineer
AI/ML Engineer developing AI solutions for professional services firms. Working on production models and real data to enhance client relationships.