# waymo-research/waymax

A JAX-based simulator for autonomous driving research.

Repository: https://github.com/waymo-research/waymax
Canonical: https://ross.abutalabs.com/products/waymax
Language: Python
License: NOASSERTION
License Family: other
Last push: 2025-10-23T04:29:20+00:00

## Health v2 (maintenance only)
Score: 49/100 (v2, computed 2026-09-03T02:20:16.233290+00:00)
- activity 48, release rhythm 35, longevity 76
- inputs: {"age_days": 1066, "days_push": 314, "days_rel": null, "gap_med": null, "n_releases_24m": 0}
- flags: no_releases, no_license
- formula: round(0.45*activity + 0.35*rhythm + 0.20*longevity); archived -> min(score, 10)

## Adoption (not part of the score)
Stars 1096, forks 132 (observed 2026-08-28T04:03:34.315824+00:00)

## What it is
Waymax is a lightweight, multi-agent autonomous driving simulator built entirely in JAX, based on the Waymo Open Motion Dataset. It supports closed-loop planning simulation, sim agent research, and open-loop behavior prediction using bounding-box object representations, and runs efficiently on GPUs and TPUs.

## Use cases
- simulate autonomous driving scenarios for research
- train and evaluate planning policies in closed-loop simulation
- run behavior prediction experiments on the Waymo Open Motion Dataset
- build learned sim agents for driving simulation
- benchmark driving policies with common metrics and RL interfaces
- accelerate driving simulation on GPUs and TPUs with JAX

## When to choose
- you need a fast, JAX-accelerated multi-agent driving simulator
- your research uses the Waymo Open Motion Dataset
- you want closed-loop evaluation for planning or sim agents
- you need GPU/TPU-distributed simulation with RL environment adapters

## When to avoid
- you need raw sensor data simulation rather than bounding-box abstractions
- you want a commercial-use simulator (license is non-commercial only)
- you need a full autonomous driving stack with perception and sensor simulation
- you don't work in Python or the JAX ecosystem

## Facets
- artifact type: library
- maturity: active
- function: simulation, machine-learning, reinforcement-learning, data-science
- domain: autonomous-vehicles, machine-learning, robotics, simulation
- platform: python
- tags: jax, autonomous-driving, waymo-open-motion-dataset, behavior-prediction, closed-loop-simulation, sim-agents, motion-forecasting, gpu, linux, macos

## Member repositories
- waymo-research/waymax (main) score 49

## Provenance
- Observed fields: from GitHub, fetched 2026-08-28T04:03:34.315824+00:00.
- Health v2: computed from the inputs above; adoption is never an input.
- Inferred fields (summary, facets, guidance): AI-extracted, prompt v1, taxonomy v1, on 2026-08-30T06:47:12.691653+00:00, confidence not recorded.
  - readme: https://github.com/waymo-research/waymax (fetched 2026-08-28T04:03:34.315824+00:00, sha 05dd03669534)
- Data as of 2026-08-30T08:39:29.467469+00:00.