Research Engineer, World Models
Core
Develop algorithms and productionize next-generation World Models for temporal reasoning and generation in complex, dynamic 4D environments to power autonomous driving simulation.
Role type
Senior IC research engineer (generative world models)
Builds
Waabi World simulation platform for autonomous trucks and robotaxis
Domain
Autonomous driving, Physical AI, Generative AI
Deliverable
production ML models
Required skills
Python, PyTorch (or JAX), distributed training, large-scale model deployment, generative modeling, predictive modeling, multimodal foundation models, LLM/VLM/VLA methods, model distillation, data pipeline engineering
Preferred skills
Kubeflow, Ray, ONNX, TensorRT, cloud platforms
Responsibilities
Design and scale state-of-the-art generative and predictive world-modeling systems; collaborate with Research Scientists to translate prototypes into robust training/inference pipelines; optimize model training and inference for efficiency; build large-scale data pipelines; ensure codebase quality and stability; stay current on advances in generative AI and robotics deployment
Seniority
Senior, hands-on IC