Research Scientist, World Models jobs in United States
cer-icon
Apply on Employer Site
company-logo

Waabi · 1 month ago

Research Scientist, World Models

Waabi is an AI company building the next generation of self-driving technology. As a Research Scientist in World Models, you will develop algorithms for World Simulation models that can reason about complex, dynamic 4D environments with minimal supervision, directly impacting Waabi World’s capabilities.

Artificial Intelligence (AI)AutomotiveAutonomous VehiclesMachine Learning
check
H1B Sponsor Likelynote

Responsibilities

Conduct fundamental and applied research in generative and predictive world-modeling (e.g., latent diffusion, video generation models, 3D Gaussian splatting, NeRF) on multi-sensor multi-modal data of driving scenes
Collaborate with simulation engineers to integrate models into large-scale, distributed training and rendering pipelines
Publish high-impact research at top conferences (CVPR, ECCV, ICCV, NeurIPS, ICLR, ICRA, SIGGRAPH)
Mentor junior scientists and interns; foster a culture of scientific rigor and rapid experimentation
Stay on top of emerging advances in generative AI, differentiable rendering, knowledge distillation/compression, and robotics

Qualification

Ph.D. in Computer VisionPythonPyTorchGenerative modelingDistributed trainingMachine LearningTeam playerCommunication

Required

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

Preferred

Proven ability to translate research into production-quality code and measurable product impact
Demonstrated publications (first-author) in top-tier venues on topics such as world models, generative simulation, video prediction, NeRF/3DGS, diffusion, flow-matching, or foundation models for autonomy

Benefits

Competitive compensation and equity awards.
Health and Wellness benefits encompassing Medical, Dental and Vision coverage (for full-time employees only).
Unlimited Vacation.
Flexible hours and Work from Home support.
Daily drinks, snacks and catered meals (when in office).
Regularly scheduled team building activities and social events both on-site, off-site & virtually.

Company

Waabi

twittertwittertwitter
company-logo
Waabi is pioneering Physical AI starting with autonomous trucking.

H1B Sponsorship

Waabi has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (8)
2024 (6)
2023 (6)
2022 (4)

Funding

Current Stage
Growth Stage
Total Funding
$283.04M
Key Investors
Volvo Group Venture CapitalKhosla Ventures
2024-06-18Series B· $200.48M
2023-01-18Series Unknown
2021-06-08Series A· $82.55M

Leadership Team

leader-logo
Raquel Urtasun
Founder & CEO
linkedin
leader-logo
Lior Ron
COO
linkedin
Company data provided by crunchbase