Research Engineer, World Models

Job summary

Responsibilities

With offices in Toronto, San Francisco, Dallas, and Pittsburgh, Waabi is growing quickly and looking for diverse, innovative and collaborative candidates who want to impact the world in a positive way. To learn more visit: www.waabi.ai

• Design, implement, and scale state-of-the-art generative and predictive world-modeling systems:
• Video generation and prediction
• Latent diffusion / autoregressive / flow-matching models
• Multimodal foundation models for driving scenes
• LLM / VLM / VLA methods for scene understanding, reasoning, and control

What they're looking for

• Strong software engineering and implementation: You have very strong Python & PyTorch (or JAX) skills; strong software-engineering fundamentals, and extensive experience with…
• Demonstrated technical impact: You have a Master's degree in Computer Vision, Machine Learning, Robotics, or a related field, or equivalent industry experience in model…
• Expert domain knowledge: You have built and deployed generative or predictive models of the physical world, focusing on scale, efficiency, and robustness for real-world…
• Team player: You have worked in a close-knit team of researchers and engineers and have strong communication to deliver successful projects in a fast-paced environment.

Tech stack

• Python
• PyTorch
• JAX
• Kubeflow
• Ray