Senior / Staff Research Engineer, Waabi

• Experience with imitation learning, reinforcement learning, planning, and/or robotics
• Expertise with ML frameworks such as PyTorch, TensorFlow, or Jax
• Proven ability to take research from prototype to production systems
• Excellent software engineering skills with experience building scalable ML systems
• Open-minded and collaborative team player with willingness to help others

Desirable

• The Behaviors team at Waabi develops cutting-edge simulation agents and scenario generation algorithms for Waabi World, our simulation platform
• As a Senior/Staff Research Engineer on our Behaviors team, you will create the next generation of simulation software for autonomous driving
• You will work closely with our multidisciplinary team of research scientists and software engineers to build innovative, practical, and scalable solutions to challenging problems as part of our next-generation self-driving solution
• Architect robust and scalable training and evaluation frameworks for IL/RL-based traffic agents
• Design and implement scalable data pipelines for generating large-scale training datasets
• Productionize cutting-edge research into robust, scalable systems in Waabi World, our simulation platform
• Contribute to technical direction and strategy for realistic traffic agents and scenario generation