Research Scientist, 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

• Conduct fundamental and applied research in generative and predictive world-modeling
• 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

• Demonstrated technical innovation: You have a Ph.D. in Computer Vision, Machine Learning, Robotics, or a related field or equivalent research experience pushing the boundaries of…
• Strong prototyping and implementation: You have expert-level Python & PyTorch (or JAX) skills; strong software-engineering fundamentals and experience with distributed training.
• Expert domain knowledge: You have built generative or predictive models of the physical world with scale and efficiency in mind for real-world applications
• Team player: You have worked in a close-knit team of researchers and engineers and have strong communication to deliver successful projects.

Tech stack

• Python
• PyTorch
• JAX