GPUDrive: Data-driven, multi-agent driving simulation at 1 million FPS

https://arxiv.org/abs/2408.01584

Built on Madrona; over one million simulation steps per second; multi-agent driving agents trained on Waymo Open Motion scenarios in minutes to hours.

Massively parallel GPU simulation