San Francisco, CA Hybrid Employment $155,000 - $269,000

Waabi is hiring a Research Scientist, World Models

About the Role

The role involves leading research in world models to enhance autonomous vehicle simulation. Responsibilities include designing machine learning frameworks that model complex environments and improving system accuracy through scalable, data-driven approaches.

Responsibilities

  • Lead research initiatives in world model development
  • Train and evaluate generative models for environment simulation
  • Improve fidelity of synthetic data generation
  • Work closely with software engineers to deploy models
  • Analyze system-level impact of model improvements
  • Optimize training pipelines for efficiency and scalability
  • Develop metrics for assessing simulation realism
  • Collaborate across disciplines including perception and planning
  • Stay current with advancements in generative modeling
  • Contribute to technical strategy and roadmap

Compensation

Competitive salary and equity package

Work Arrangement

Hybrid work model with flexibility based on team and project needs

Team

Part of a focused research team working on core simulation and prediction technologies

What You'll Do

  • Develop next-generation world models using deep learning and probabilistic methods
  • Design simulation environments that accurately reflect real-world driving scenarios
  • Collaborate with engineering teams to integrate research into production systems
  • Evaluate model performance under diverse and edge-case conditions
  • Publish findings in top-tier conferences and journals

What We're Looking For

  • PhD in computer science, robotics, machine learning, or related technical field
  • Strong background in deep learning, especially sequence modeling and generative models
  • Experience with large-scale simulation systems
  • Proficiency in Python and machine learning frameworks such as PyTorch or TensorFlow
  • Proven research record with publications in relevant venues

Bonus Qualifications

  • Experience with autonomous vehicles or robotics platforms
  • Familiarity with differentiable physics simulation
  • Knowledge of sensor modeling for cameras, lidar, and radar
  • Contributions to open-source simulation or ML projects
  • Industry experience in deploying ML models at scale

Visa sponsorship available for qualified candidates

About company
Waabi
Waabi appears to be a technology company focused on autonomous vehicle systems engineering.
All jobs at Waabi Visit website
Job Details
Department Software – Autonomy & Algorithms
Category other
Posted 2 months ago