Waymo

Waymax Driving Simulator

Waymax is a JAX-based, lightweight, multi-agent simulator for autonomous driving research, designed for accelerated training and evaluation of driving policies. Open-sourced by Waymo Research and used to benchmark behaviour-prediction and planning models against logged Waymo Open Motion Dataset scenarios.

Waymax Driving Simulator is one of 2 APIs that Waymo publishes on the APIs.io network.

Tagged areas include Simulator, JAX, Reinforcement Learning, and Research.

GitHub

Other Resources

🔗
Repository
https://github.com/waymo-research/waymax

API entry from apis.yml

apis.yml Raw ↑ 
aid: waymo:waymax
name: Waymax Driving Simulator
description: Waymax is a JAX-based, lightweight, multi-agent simulator for autonomous driving research,
  designed for accelerated training and evaluation of driving policies. Open-sourced by Waymo Research
  and used to benchmark behaviour-prediction and planning models against logged Waymo Open Motion Dataset
  scenarios.
humanURL: https://github.com/waymo-research/waymax
baseURL: https://github.com/waymo-research/waymax
tags:
- Simulator
- JAX
- Reinforcement Learning
- Research
properties:
- type: Repository
  url: https://github.com/waymo-research/waymax