- Lightweight – It enables researchers to conduct experiments efficiently and iterate quickly;
- Informed by troves of real-world data – To model realistic driving behaviors, Waymax can leverage the 570+ hours of real-world driving examples from multiple cities that are present in the Waymo Open Motion Dataset;
- Hardware-accelerated – Waymax is written entirely in JAX, making use of in-graph compilation to deliver blazing-fast simulations;
- Multi-agent – so researchers can analyze and model the complex behaviors among multiple road users and unlock new insights into multi-agent systems.
Researchers can access Waymax software on GitHub and use it straight out of the box. It already incorporates the minimal representation of objects and bounding boxes adopted by the Waymo Open Motion Dataset, which eliminates the need for users to run their own perception system from raw sensor data, and comes bundled with metrics and pre-built sim agents, streamlining the research process.