Waymo announces Waymax – a lightweight simulator for autonomous driving research