Tech Lead, Wayve Labs

Wayve Wayve · Robotics · Sunnyvale, CA · Wayve Labs

The role focuses on designing and training large-scale embodied foundation models for autonomous driving, developing world models and planners, innovating in reinforcement learning and reward modeling, and conducting empirical research on scaling laws and generalization. It involves working at the intersection of machine learning, simulation, robotics, and real-world deployment.

What you'd actually do

  1. Design and train large-scale embodied foundation models, advancing architectures in attention, mixture-of-experts, and distributed training.
  2. Develop world models and planners (e.g., diffusion, transformer-based, or hybrid generative approaches) to simulate diverse and temporally consistent driving environments.
  3. Innovate in reinforcement learning and reward modeling, building safe, efficient, and scalable reward frameworks integrated with real-world and synthetic data.
  4. Conduct empirical research on scaling laws, generalisation, and synthetic-to-real transfer for autonomous systems.
  5. Define benchmarks and metrics for long-horizon prediction, scene fidelity, planner integration, and driving performance.

Skills

Required

  • Python
  • PyTorch
  • Machine Learning
  • Computer Vision
  • Robotics

Nice to have

  • autonomous driving
  • simulation
  • FSDP
  • DeepSpeed
  • JAX
  • synthetic-to-real transfer
  • data-efficient learning
  • open-source ML tools
  • research infrastructure

What the JD emphasized

  • PhD, Master’s degree, or equivalent experience in Machine Learning, Computer Vision, Robotics, or related fields
  • minimum of 4 years’ experience in applied ML research or engineering
  • Track record of publishing at top-tier conferences (e.g., NeurIPS, ICML, ICLR, CVPR, ICCV, CoRL)

Other signals

  • design and train large-scale embodied foundation models
  • develop world models and planners
  • innovate in reinforcement learning and reward modeling
  • conduct empirical research on scaling laws, generalisation, and synthetic-to-real transfer