Machine Learning Intern

About the role

Cantina Labs, a social AI company, is expanding its research lab in Singapore and seeks a Machine Learning Intern to contribute to next‑generation video generation models. The three‑month onsite internship starts in October 2026 (or January 2027) and offers ownership of a well‑defined research or engineering problem under a senior mentor.

Key responsibilities

• Develop reward models to improve aesthetics, motion quality, temporal consistency, and prompt adherence.
• Study base‑model behavior and use findings to inform post‑training outcomes.
• Design rigorous evaluations and conduct large‑scale experiments on generative video models.
• Build systems for ingesting, preprocessing, curating, and delivering large‑scale video datasets.
• Develop distributed pipelines for dataset generation, deduplication, and refreshes.
• Improve reliability, reproducibility, and efficiency of data and model‑training workflows.
• Build tooling for video and multimodal data using FFmpeg, PyAV, DALI, or OpenCV.
• Contribute to evaluation harnesses, model integrations, and research tooling.

Required profile

• Pursuing a bachelor’s or master’s degree in computer science, engineering, machine learning, or a related field.
• Experience building data pipelines, ML infrastructure, or distributed systems through research, coursework, open‑source contributions, or prior internships.
• Familiarity with tools such as Ray, PySpark, Airflow, Docker, Kubernetes, and cloud platforms (AWS, Google Cloud, Azure).
• Understanding of data throughput, storage layout, caching, monitoring, and failure recovery.
• Proficiency in Python and interest in reliable large‑scale machine learning systems.

Required skills

• Python
• Ray
• PySpark
• Airflow
• Docker
• Kubernetes
• AWS
• Google Cloud
• Azure
• FFmpeg
• PyAV
• DALI
• OpenCV

What we offer

• A defined project and senior mentor from day one.
• Weekly one‑on‑one meetings and clear milestones.
• Compute allocation for research and potential first‑author publication.
• Support for conference travel if a paper is accepted.
• Competitive monthly stipend.
• Visa and travel support for eligible international candidates.
• Housing support for qualifying interns in Singapore.
• Equipment and resources needed for the work.