Distributed training and large-scale model deployment
Generative or predictive models of the physical world
Waabi is the leader in Physical AI, unlocking the next era of autonomous transportation with technology powering commercial autonomous trucks and robotaxis
Job Summary
Waabi is the leader in Physical AI, unlocking the next era of autonomous transportation with technology powering commercial autonomous trucks and robotaxis.
As a Research Engineer in the World Models team, you will develop algorithms and productionize the next generation of World Models that can reason about complex, dynamic 4D environments.
Your work will directly power Waabi World’s ability to model future evolution, synthesize realistic safety-critical scenarios, and provide rich generative priors for downstream planning, testing, and training.
Matching Summary
Waabi is the leader in Physical AI, unlocking the next era of autonomous transportation with technology powering commercial autonomous trucks and robotaxis.
Skills & Requirements
Must-have
Python & PyTorch (or JAX) skills
distributed training and large-scale model deployment
generative or predictive models of the physical world
large-scale, distributed training and inference pipelines
large scale data pipelines
Nice-to-have
infrastructure and tooling for large-scale ML training
efficient model serving and deployment
publications or research at top ML/CV/Robotics conference
Key Requirements
Master's degree or equivalent industry experience
strong software-engineering fundamentals
extensive experience with distributed training
experience with large-scale model deployment
built and deployed generative or predictive models