Research Engineer - World Modeling

We are a dedicated research lab for building, understanding, using, and risk-managing foundation models. As a Research Engineer - World Modeling, you'll work on the core of cutting-edge foundation model training, alongside world-class researchers, data scientists, and engineers, tackling the most fundamental and impactful challenges in AI development.

Requirements

• MSc or PhD in Machine Learning or Computer Science, or equivalent industry experience.
• Proficient in data collection, cleaning, and transformation at scale, including designing robust pipelines for multimodal datasets (e.g., video, audio, text).
• Practical experience with web scraping and crawling frameworks (e.g., scrapy, selenium, playwright, BeautifulSoup) to collect and curate high-quality web-scale datasets.
• Experience in large-scale model training (LLMs or Diffusion Models) on large clusters.
• Hands-on experience with state-of-the-art video generative models (e.g., Sora, Veo2, MovieGen, CogVideoX, etc.).
• Experiences in building and optimizing large-scale video data pipelines.
• Experience in accelerating diffusion model inference for improved efficiency.
• Exceptional problem-solving and troubleshooting skills to tackle complex technical challenges.
• Strong systems and engineering expertise in deep learning frameworks such as PyTorch.
• Strong communication and collaboration skills for effective cross-functional teamwork.
• Demonstrated ability to solve complex system-level challenges and debug failures across the training/inference stack (e.g., memory issues, deadlocks, I/O bottlenecks).

Benefits

• Comprehensive medical, dental, and vision benefits
• Bonus
• 401K Plan
• Generous paid time off, sick leave and holidays
• Paid Parental Leave
• Employee Assistance Program
• Life insurance and disability