Shypd.aiThe largest open index of AI toolsWaymax is a JAX-based simulator for autonomous driving research. It provides a lightweight, multi-agent simulation environment based on the Waymo Open Motion Dataset.
Trending
About
Waymax is a lightweight, multi-agent, JAX-based simulator specifically designed for autonomous driving research. Built upon the Waymo Open Motion Dataset, it supports various aspects of behavior research, from closed-loop simulation for planning and sim agent development to open-loop behavior prediction. The simulator represents objects like vehicles and pedestrians as bounding boxes, simplifying behavior research. Its JAX-based architecture ensures easy distribution and deployment on hardware accelerators such as GPUs and TPUs. Waymax includes libraries for data loading, visualization, metric computation, intelligent sim agents, and adapters for common RL interfaces like dm-env, allowing for both standalone module use and full closed-loop simulation.
Capabilities
Pricing & Plans
Open Source
Free
FAQs
Trending
