About code release for Waymax simulator

Hi,

Thank you for the code for autonomous racing. 

As I am working on a similar project, can you kindly let me know if you are planning on releasing the code for waymax related simulations? I would be very thankful as it would greatly help the benchmarking of the project.

Thanks