Index

_

__eq__() (waymax.datatypes.Action method) (waymax.datatypes.action.Action method) (waymax.datatypes.action.TrajectoryUpdate method) (waymax.datatypes.array.MaskedArray method) (waymax.datatypes.MaskedArray method) (waymax.datatypes.object_state.ObjectMetadata method) (waymax.datatypes.object_state.Trajectory method) (waymax.datatypes.ObjectMetadata method) (waymax.datatypes.ObjectPose2D method) (waymax.datatypes.observation.ObjectPose2D method) (waymax.datatypes.Paths method) (waymax.datatypes.roadgraph.RoadgraphPoints method) (waymax.datatypes.RoadgraphPoints method) (waymax.datatypes.route.Paths method) (waymax.datatypes.simulator_state.SimulatorState method) (waymax.datatypes.SimulatorState method) (waymax.datatypes.traffic_lights.TrafficLights method) (waymax.datatypes.TrafficLights method) (waymax.datatypes.Trajectory method) (waymax.datatypes.TrajectoryUpdate method) (waymax.env.wrappers.brax_wrapper.TimeStep method) __post_init__() (waymax.config.EnvironmentConfig method) (waymax.config.WaymaxConfig method) _ActorCore (in module waymax.agents.actor_core) _add_headway_waypoints() (in module waymax.agents.waypoint_following_agent) _clip_values() (waymax.dynamics.bicycle_model.InvertibleBicycleModel method) (waymax.dynamics.delta.DeltaGlobal method) (waymax.dynamics.DeltaGlobal method) (waymax.dynamics.InvertibleBicycleModel method) _compute_arclengths() (in module waymax.agents.waypoint_following_agent) _compute_lead_distance() (waymax.agents.IDMRoutePolicy method) (waymax.agents.waypoint_following_agent.IDMRoutePolicy method) _compute_lead_velocity() (waymax.agents.IDMRoutePolicy method) (waymax.agents.waypoint_following_agent.IDMRoutePolicy method) _DEFAULT_CONTROL_FUNC (in module waymax.agents.sim_agent) _DEFAULT_LEAD_DISTANCE (in module waymax.agents.waypoint_following_agent) _DEFAULT_LEAD_VELOCITY (in module waymax.agents.waypoint_following_agent)

_DEFAULT_TIME_DELTA (in module waymax.agents.waypoint_following_agent) _DISTANCE_TO_REF_THRESHOLD (in module waymax.agents.waypoint_following_agent) _EXPERT_NAME (in module waymax.agents.expert) _find_reference_traj_from_log_traj() (in module waymax.agents.waypoint_following_agent) _get_accel() (waymax.agents.IDMRoutePolicy method) (waymax.agents.waypoint_following_agent.IDMRoutePolicy method) _get_control_mask() (in module waymax.env.base_environment) _get_invalid_future_traffic_light() (in module waymax.dataloader.womd_utils) _get_invalid_future_trajectory() (in module waymax.dataloader.womd_utils) _get_next_trajectory_by_projection() (waymax.agents.waypoint_following_agent.WaypointFollowingPolicy method) (waymax.agents.WaypointFollowingPolicy method) _index_pytree() (in module waymax.visualization.viz) _initialized_overlap() (in module waymax.env.planning_agent_environment) _INVALID_FLOAT_VALUE (in module waymax.datatypes.object_state) _INVALID_INT_VALUE (in module waymax.datatypes.object_state) _IS_SDC_FUNC (in module waymax.agents.expert) _KIM_EPSILON (in module waymax.metrics.comfort) _linear_config_to_metric_config() (in module waymax.rewards.linear_combination_reward) _METRICS_REGISTRY (in module waymax.metrics.metric_factory) _MINIMUM_LEAD_DISTANCE (in module waymax.agents.waypoint_following_agent) _multi_agent_reward_spec() (in module waymax.env.base_environment) _object_metadata_to_dict() (in module waymax.dataloader.womd_utils) _plot_bounding_boxes() (in module waymax.visualization.viz) _plot_path_points() (in module waymax.visualization.viz) _project_to_a_trajectory() (in module waymax.agents.waypoint_following_agent) _REACHED_END_OF_TRAJECTORY_THRESHOLD (in module waymax.agents.waypoint_following_agent) _roadgraph_to_dict() (in module waymax.dataloader.womd_utils) _RoadGraphDefaultColor (in module waymax.visualization.viz) _RoadGraphShown (in module waymax.visualization.viz) _SPEED_LIMIT (in module waymax.dynamics.bicycle_model) _STATIC_SPEED_THRESHOLD (in module waymax.agents.waypoint_following_agent) _TF_TO_JNP_DTYPE (in module waymax.dataloader.womd_utils) _traffic_light_to_dict() (in module waymax.dataloader.womd_utils) _trajectory_to_dict() (in module waymax.dataloader.womd_utils) _transform_points() (in module waymax.utils.geometry) _tree_expand_and_repeat() (in module waymax.datatypes.observation) _validate_reward_metrics() (in module waymax.rewards.linear_combination_reward) _vmap_over_batch_dims() (in module waymax.datatypes.operations)

A

AbstractEnvironment (class in waymax.env) (class in waymax.env.abstract_environment) AbstractMetric (class in waymax.metrics) (class in waymax.metrics.abstract_metric) AbstractRewardFunction (class in waymax.rewards) (class in waymax.rewards.abstract_reward_function) Action (class in waymax.datatypes) (class in waymax.datatypes.action) (in module waymax.agents.actor_core) action (waymax.agents.actor_core.WaymaxActorOutput attribute), [1] (waymax.agents.WaymaxActorOutput attribute), [1] (waymax.env.rollout.RolloutOutput attribute), [1] (waymax.env.RolloutOutput attribute), [1] action_spec() (waymax.dynamics.abstract_dynamics.DynamicsModel method) (waymax.dynamics.bicycle_model.InvertibleBicycleModel method) (waymax.dynamics.delta.DeltaGlobal method) (waymax.dynamics.DeltaGlobal method) (waymax.dynamics.DiscreteActionSpaceWrapper method) (waymax.dynamics.discretizer.DiscreteActionSpaceWrapper method) (waymax.dynamics.DynamicsModel method) (waymax.dynamics.InvertibleBicycleModel method) (waymax.dynamics.state_dynamics.StateDynamics method) (waymax.dynamics.StateDynamics method) (waymax.env.abstract_environment.AbstractEnvironment method) (waymax.env.AbstractEnvironment method) (waymax.env.base_environment.BaseEnvironment method) (waymax.env.BaseEnvironment method) (waymax.env.BraxWrapper method) (waymax.env.DMEnvWrapper method) (waymax.env.planning_agent_environment.PlanningAgentDynamics method) (waymax.env.planning_agent_environment.PlanningAgentEnvironment method) (waymax.env.PlanningAgentDynamics method) (waymax.env.PlanningAgentEnvironment method) (waymax.env.wrappers.brax_wrapper.BraxWrapper method) (waymax.env.wrappers.dm_env_wrapper.DMEnvWrapper method)

actor_core_factory() (in module waymax.agents) (in module waymax.agents.actor_core) actor_state (waymax.agents.actor_core.WaymaxActorOutput attribute), [1] (waymax.agents.WaymaxActorOutput attribute), [1] (waymax.env.rollout.RolloutCarry attribute), [1] ActorState (in module waymax.agents.actor_core) agent_type (waymax.config.SimAgentConfig attribute), [1] aggregate_time_tensors() (in module waymax.dataloader.womd_utils) aggregate_timesteps (waymax.config.DatasetConfig attribute), [1] allow_new_objects_after_warmup (waymax.config.EnvironmentConfig attribute), [1] apply_trajectory_update_to_state() (in module waymax.dynamics.abstract_dynamics) arc_length (waymax.datatypes.Paths attribute), [1] (waymax.datatypes.route.Paths attribute), [1] ArrayLike (in module waymax.datatypes.operations) (in module waymax.datatypes.simulator_state) ARROW_CAUTION (waymax.datatypes.traffic_lights.TrafficLightStates attribute) (waymax.datatypes.TrafficLightStates attribute) ARROW_GO (waymax.datatypes.traffic_lights.TrafficLightStates attribute) (waymax.datatypes.TrafficLightStates attribute) ARROW_STOP (waymax.datatypes.traffic_lights.TrafficLightStates attribute) (waymax.datatypes.TrafficLightStates attribute) as_action() (waymax.datatypes.action.TrajectoryUpdate method) (waymax.datatypes.TrajectoryUpdate method) AUTOTUNE (in module waymax.dataloader.dataloader_utils) (in module waymax.dataloader.womd_dataloader) Axis (in module waymax.datatypes.operations)

B

back_x (waymax.visualization.utils.VizConfig attribute) back_y (waymax.visualization.utils.VizConfig attribute) BaseEnvironment (class in waymax.env) (class in waymax.env.base_environment) batch_by_scenario (waymax.config.DatasetConfig attribute), [1] batch_dims (waymax.config.DatasetConfig attribute), [1] (waymax.datatypes.Observation property) (waymax.datatypes.observation.Observation property) (waymax.datatypes.simulator_state.SimulatorState property) (waymax.datatypes.SimulatorState property)

bbox_corners (waymax.datatypes.object_state.Trajectory property) (waymax.datatypes.Trajectory property) BraxWrapper (class in waymax.env) (class in waymax.env.wrappers.brax_wrapper)

C

CAUTION (waymax.datatypes.traffic_lights.TrafficLightStates attribute) (waymax.datatypes.TrafficLightStates attribute) center_agent_idx (waymax.visualization.utils.VizConfig attribute) center_at_xy() (in module waymax.visualization.utils) COLOR_DICT (in module waymax.visualization.color) combine_two_object_pose_2d() (in module waymax.datatypes.observation) compare_all_leaf_nodes() (in module waymax.datatypes) (in module waymax.datatypes.operations) compute() (waymax.metrics.abstract_metric.AbstractMetric method) (waymax.metrics.AbstractMetric method) (waymax.metrics.comfort.KinematicsInfeasibilityMetric method) (waymax.metrics.imitation.LogDivergenceMetric method) (waymax.metrics.LogDivergenceMetric method) (waymax.metrics.OffroadMetric method) (waymax.metrics.OffRouteMetric method) (waymax.metrics.overlap.OverlapMetric method) (waymax.metrics.OverlapMetric method) (waymax.metrics.ProgressionMetric method) (waymax.metrics.roadgraph.OffroadMetric method) (waymax.metrics.roadgraph.WrongWayMetric method) (waymax.metrics.route.OffRouteMetric method) (waymax.metrics.route.ProgressionMetric method) (waymax.metrics.WrongWayMetric method) (waymax.rewards.abstract_reward_function.AbstractRewardFunction method) (waymax.rewards.AbstractRewardFunction method) (waymax.rewards.linear_combination_reward.LinearCombinationReward method) (waymax.rewards.LinearCombinationReward method) compute_inverse() (in module waymax.dynamics.bicycle_model) compute_kinematics_infeasibility() (waymax.metrics.comfort.KinematicsInfeasibilityMetric method) compute_log_divergence() (waymax.metrics.imitation.LogDivergenceMetric class method) (waymax.metrics.LogDivergenceMetric class method) compute_overlap() (waymax.metrics.overlap.OverlapMetric method) (waymax.metrics.OverlapMetric method) compute_pairwise_overlaps() (in module waymax.utils.geometry) compute_reward (waymax.config.EnvironmentConfig attribute), [1] compute_signed_distance_to_nearest_road_edge_point() (in module waymax.metrics.roadgraph) compute_update() (waymax.dynamics.abstract_dynamics.DynamicsModel method) (waymax.dynamics.bicycle_model.InvertibleBicycleModel method) (waymax.dynamics.delta.DeltaGlobal method) (waymax.dynamics.delta.DeltaLocal method) (waymax.dynamics.DeltaGlobal method) (waymax.dynamics.DeltaLocal method) (waymax.dynamics.DiscreteActionSpaceWrapper method) (waymax.dynamics.discretizer.DiscreteActionSpaceWrapper method) (waymax.dynamics.DynamicsModel method) (waymax.dynamics.InvertibleBicycleModel method) (waymax.dynamics.state_dynamics.StateDynamics method) (waymax.dynamics.StateDynamics method) (waymax.env.planning_agent_environment.PlanningAgentDynamics method) (waymax.env.PlanningAgentDynamics method)

config (waymax.env.DMEnvWrapper property) (waymax.env.wrappers.dm_env_wrapper.DMEnvWrapper property) ConstantSpeedPolicy (class in waymax.agents.constant_speed) CONTROLLABLE_FIELDS (in module waymax.dynamics.abstract_dynamics) controlled_object (waymax.config.EnvironmentConfig attribute), [1] controlled_objects (waymax.config.SimAgentConfig attribute), [1] coordinate_frame (waymax.config.ObservationConfig attribute), [1] CoordinateFrame (class in waymax.config) corners_from_bbox() (in module waymax.utils.geometry) corners_from_bboxes() (in module waymax.utils.geometry) create_and_validate() (waymax.datatypes.array.MaskedArray class method) (waymax.datatypes.MaskedArray class method) create_constant_speed_actor() (in module waymax.agents) (in module waymax.agents.constant_speed) create_expert_actor() (in module waymax.agents) (in module waymax.agents.expert) create_invalid_traffic_lights() (in module waymax.utils.test_utils) create_metadata() (in module waymax.utils.test_utils) create_sim_agents_from_config() (in module waymax.agents) (in module waymax.agents.agent_builder) create_test_map_element() (in module waymax.utils.test_utils) create_test_trajectory_from_position() (in module waymax.utils.test_utils) CROSSWALK (waymax.datatypes.MapElementIds attribute) (waymax.datatypes.roadgraph.MapElementIds attribute) current_log_trajectory (waymax.datatypes.simulator_state.SimulatorState property) (waymax.datatypes.SimulatorState property) current_sim_trajectory (waymax.datatypes.simulator_state.SimulatorState property) (waymax.datatypes.SimulatorState property) CYCLIST (waymax.datatypes.object_state.ObjectTypeIds attribute) (waymax.datatypes.ObjectTypeIds attribute)

D

data (waymax.datatypes.Action attribute), [1] (waymax.datatypes.action.Action attribute), [1] data_config (waymax.config.WaymaxConfig attribute), [1] data_format (waymax.config.DatasetConfig attribute), [1] DataFormat (class in waymax.config) DatasetConfig (class in waymax.config) DEFAULT_BOOL (in module waymax.dataloader.womd_utils) DEFAULT_FLOAT (in module waymax.dataloader.womd_utils) DEFAULT_INT (in module waymax.dataloader.womd_utils) delta_yaw (waymax.datatypes.ObjectPose2D attribute), [1] (waymax.datatypes.observation.ObjectPose2D attribute), [1] DeltaGlobal (class in waymax.dynamics) (class in waymax.dynamics.delta) DeltaLocal (class in waymax.dynamics) (class in waymax.dynamics.delta) deterministic (waymax.config.DatasetConfig attribute), [1] dir_x (waymax.datatypes.roadgraph.RoadgraphPoints attribute), [1] (waymax.datatypes.RoadgraphPoints attribute), [1] dir_xy (waymax.datatypes.roadgraph.RoadgraphPoints property) (waymax.datatypes.RoadgraphPoints property) dir_xyz (waymax.datatypes.roadgraph.RoadgraphPoints property) (waymax.datatypes.RoadgraphPoints property) dir_y (waymax.datatypes.roadgraph.RoadgraphPoints attribute), [1] (waymax.datatypes.RoadgraphPoints attribute), [1] dir_z (waymax.datatypes.roadgraph.RoadgraphPoints attribute), [1] (waymax.datatypes.RoadgraphPoints attribute), [1] discount (waymax.env.wrappers.brax_wrapper.TimeStep attribute), [1] discount_spec() (waymax.env.abstract_environment.AbstractEnvironment method) (waymax.env.AbstractEnvironment method) (waymax.env.base_environment.BaseEnvironment method) (waymax.env.BaseEnvironment method) (waymax.env.BraxWrapper method) (waymax.env.DMEnvWrapper method) (waymax.env.planning_agent_environment.PlanningAgentEnvironment method) (waymax.env.PlanningAgentEnvironment method) (waymax.env.wrappers.brax_wrapper.BraxWrapper method) (waymax.env.wrappers.dm_env_wrapper.DMEnvWrapper method)

DiscreteActionSpaceWrapper (class in waymax.dynamics) (class in waymax.dynamics.discretizer) discretize() (waymax.dynamics.Discretizer method) (waymax.dynamics.discretizer.Discretizer method) Discretizer (class in waymax.dynamics) (class in waymax.dynamics.discretizer) distributed (waymax.config.DatasetConfig attribute), [1] DMEnvWrapper (class in waymax.env) (class in waymax.env.wrappers.dm_env_wrapper) done (waymax.env.wrappers.brax_wrapper.TimeStep attribute), [1] drop_remainder (waymax.config.DatasetConfig attribute), [1] dynamic_index() (in module waymax.datatypes) (in module waymax.datatypes.operations) dynamic_slice() (in module waymax.datatypes) (in module waymax.datatypes.operations) dynamic_update_slice_in_dim (in module waymax.datatypes) (in module waymax.datatypes.operations) dynamics (waymax.env.base_environment.BaseEnvironment property) (waymax.env.BaseEnvironment property) (waymax.env.planning_agent_environment.PlanningAgentEnvironment property) (waymax.env.PlanningAgentEnvironment property) DynamicsModel (class in waymax.dynamics) (class in waymax.dynamics.abstract_dynamics) (in module waymax.dynamics.bicycle_model)

E

env_config (waymax.config.WaymaxConfig attribute), [1] EnvironmentConfig (class in waymax.config)

EpisodeAlreadyFinishedError, [1] EPS (in module waymax.utils.geometry)

F

F (in module waymax.datatypes.operations) fill_invalid_trajectory() (in module waymax.datatypes) (in module waymax.datatypes.object_state) filter_topk_roadgraph_points() (in module waymax.datatypes) (in module waymax.datatypes.roadgraph) FLASHING_CAUTION (waymax.datatypes.traffic_lights.TrafficLightStates attribute) (waymax.datatypes.TrafficLightStates attribute) FLASHING_STOP (waymax.datatypes.traffic_lights.TrafficLightStates attribute) (waymax.datatypes.TrafficLightStates attribute) for_obj() (waymax.datatypes.Observation method) (waymax.datatypes.observation.Observation method) forward() (waymax.dynamics.abstract_dynamics.DynamicsModel method) (waymax.dynamics.DynamicsModel method) (waymax.env.planning_agent_environment.PlanningAgentDynamics method) (waymax.env.PlanningAgentDynamics method)

from_center_and_yaw() (waymax.datatypes.ObjectPose2D class method) (waymax.datatypes.observation.ObjectPose2D class method) from_transformation() (waymax.datatypes.ObjectPose2D class method) (waymax.datatypes.observation.ObjectPose2D class method) front_x (waymax.visualization.utils.VizConfig attribute) front_y (waymax.visualization.utils.VizConfig attribute) FrozenSimPolicy (class in waymax.agents) (class in waymax.agents.sim_agent) FULL_PROGRESS_VALUE (in module waymax.metrics.route)

G

generate_sharded_filenames() (in module waymax.dataloader.dataloader_utils) GenericAction (in module waymax.env.typedefs) GenericScenario (in module waymax.env.typedefs) GenericState (in module waymax.env.typedefs) get_control_mask() (in module waymax.datatypes) (in module waymax.datatypes.simulator_state) get_data_generator() (in module waymax.dataloader) (in module waymax.dataloader.dataloader_utils)

get_features_description() (in module waymax.dataloader.womd_utils) get_metric_names() (in module waymax.metrics) (in module waymax.metrics.metric_factory) get_n_colors() (in module waymax.visualization.utils) GLOBAL (waymax.config.CoordinateFrame attribute) global_observation_from_state() (in module waymax.datatypes.observation) GO (waymax.datatypes.traffic_lights.TrafficLightStates attribute) (waymax.datatypes.TrafficLightStates attribute)

H

has_overlap() (in module waymax.utils.geometry)

height (waymax.datatypes.object_state.Trajectory attribute), [1] (waymax.datatypes.Trajectory attribute), [1]

I

IDM (waymax.config.SimAgentType attribute) IDMRoutePolicy (class in waymax.agents) (class in waymax.agents.waypoint_following_agent) ids (waymax.datatypes.object_state.ObjectMetadata attribute), [1] (waymax.datatypes.ObjectMetadata attribute), [1] (waymax.datatypes.Paths attribute), [1] (waymax.datatypes.roadgraph.RoadgraphPoints attribute), [1] (waymax.datatypes.RoadgraphPoints attribute), [1] (waymax.datatypes.route.Paths attribute), [1] img_from_fig() (in module waymax.visualization.utils) include_sdc_paths (waymax.config.DatasetConfig attribute), [1] infer_expert_action() (in module waymax.agents.expert) info (waymax.env.wrappers.brax_wrapper.TimeStep attribute), [1] init() (waymax.agents.actor_core.WaymaxActorCore method) (waymax.agents.sim_agent.SimAgentActor method) (waymax.agents.SimAgentActor method) (waymax.agents.WaymaxActorCore method) init_fig_ax() (in module waymax.visualization.utils) init_fig_ax_via_size() (in module waymax.visualization.utils) init_steps (waymax.config.EnvironmentConfig attribute), [1] inverse() (waymax.dynamics.abstract_dynamics.DynamicsModel method) (waymax.dynamics.bicycle_model.InvertibleBicycleModel method) (waymax.dynamics.delta.DeltaGlobal method) (waymax.dynamics.delta.DeltaLocal method) (waymax.dynamics.DeltaGlobal method) (waymax.dynamics.DeltaLocal method) (waymax.dynamics.DiscreteActionSpaceWrapper method) (waymax.dynamics.discretizer.DiscreteActionSpaceWrapper method) (waymax.dynamics.DynamicsModel method) (waymax.dynamics.InvertibleBicycleModel method) (waymax.dynamics.state_dynamics.StateDynamics method) (waymax.dynamics.StateDynamics method) (waymax.env.planning_agent_environment.PlanningAgentDynamics method) (waymax.env.PlanningAgentDynamics method)

inverse_action_by_search() (in module waymax.dynamics.discretizer) InvertibleBicycleModel (class in waymax.dynamics) (class in waymax.dynamics.bicycle_model) is_controlled (waymax.agents.actor_core.WaymaxActorOutput attribute), [1] (waymax.agents.WaymaxActorOutput attribute), [1] (waymax.datatypes.object_state.ObjectMetadata attribute), [1] (waymax.datatypes.ObjectMetadata attribute), [1] is_done (waymax.datatypes.simulator_state.SimulatorState property) (waymax.datatypes.SimulatorState property) is_ego (waymax.datatypes.Observation attribute), [1] (waymax.datatypes.observation.Observation attribute), [1] is_modeled (waymax.datatypes.object_state.ObjectMetadata attribute), [1] (waymax.datatypes.ObjectMetadata attribute), [1] is_offroad() (in module waymax.metrics.roadgraph) is_road_edge() (in module waymax.datatypes) (in module waymax.datatypes.roadgraph) is_sdc (waymax.datatypes.object_state.ObjectMetadata attribute), [1] (waymax.datatypes.ObjectMetadata attribute), [1] is_valid (waymax.datatypes.object_state.ObjectMetadata attribute), [1] (waymax.datatypes.ObjectMetadata attribute), [1]

K

KinematicsInfeasibilityMetric (class in waymax.metrics.comfort)

L

LANE_BIKE_LANE (waymax.datatypes.MapElementIds attribute) (waymax.datatypes.roadgraph.MapElementIds attribute) LANE_FREEWAY (waymax.datatypes.MapElementIds attribute) (waymax.datatypes.roadgraph.MapElementIds attribute) lane_ids (waymax.datatypes.traffic_lights.TrafficLights attribute), [1] (waymax.datatypes.TrafficLights attribute), [1] LANE_SURFACE_STREET (waymax.datatypes.MapElementIds attribute) (waymax.datatypes.roadgraph.MapElementIds attribute) LANE_UNDEFINED (waymax.datatypes.MapElementIds attribute) (waymax.datatypes.roadgraph.MapElementIds attribute)

length (waymax.datatypes.object_state.Trajectory attribute), [1] (waymax.datatypes.Trajectory attribute), [1] LinearCombinationReward (class in waymax.rewards) (class in waymax.rewards.linear_combination_reward) LinearCombinationRewardConfig (class in waymax.config) log_traffic_light (waymax.datatypes.simulator_state.SimulatorState attribute), [1] (waymax.datatypes.SimulatorState attribute), [1] log_trajectory (waymax.datatypes.simulator_state.SimulatorState attribute), [1] (waymax.datatypes.SimulatorState attribute), [1] LogDivergenceMetric (class in waymax.metrics) (class in waymax.metrics.imitation)

M

make_continuous() (waymax.dynamics.Discretizer method) (waymax.dynamics.discretizer.Discretizer method) make_invalid_data() (in module waymax.datatypes) (in module waymax.datatypes.operations) make_sdc_dm_environment() (in module waymax.env.wrappers.dm_env_wrapper) make_test_dataset() (in module waymax.utils.test_utils) make_zeros_state() (in module waymax.utils.test_utils) MapElementIds (class in waymax.datatypes) (class in waymax.datatypes.roadgraph) masked_mean() (in module waymax.datatypes) (in module waymax.datatypes.operations) masked_value() (waymax.datatypes.array.MaskedArray method) (waymax.datatypes.MaskedArray method) MaskedArray (class in waymax.datatypes) (class in waymax.datatypes.array) matrix (waymax.datatypes.ObjectPose2D attribute), [1] (waymax.datatypes.observation.ObjectPose2D attribute), [1] MAX_DISTANCE_TO_ROUTE_PATH (waymax.metrics.OffRouteMetric attribute) (waymax.metrics.route.OffRouteMetric attribute) max_num_objects (waymax.config.DatasetConfig attribute), [1] (waymax.config.EnvironmentConfig attribute), [1] max_num_rg_points (waymax.config.DatasetConfig attribute), [1] MAX_RELATIVE_DISTANCE_TO_OFF_ROUTE_PATH (waymax.metrics.OffRouteMetric attribute) (waymax.metrics.route.OffRouteMetric attribute) merge_actions() (in module waymax.agents) (in module waymax.agents.actor_core) metadata (waymax.datatypes.Observation attribute), [1] (waymax.datatypes.observation.Observation attribute), [1] MetricResult (class in waymax.metrics) (class in waymax.metrics.abstract_metric) Metrics (in module waymax.env) (in module waymax.env.typedefs) metrics (waymax.config.EnvironmentConfig attribute), [1] (waymax.env.rollout.RolloutOutput attribute), [1] (waymax.env.RolloutOutput attribute), [1] (waymax.env.wrappers.brax_wrapper.TimeStep attribute), [1] metrics() (waymax.env.abstract_environment.AbstractEnvironment method) (waymax.env.AbstractEnvironment method) (waymax.env.base_environment.BaseEnvironment method) (waymax.env.BaseEnvironment method) (waymax.env.BraxWrapper method) (waymax.env.planning_agent_environment.PlanningAgentEnvironment method) (waymax.env.PlanningAgentEnvironment method) (waymax.env.wrappers.brax_wrapper.BraxWrapper method) metrics_to_run (waymax.config.MetricsConfig attribute), [1] MetricsConfig (class in waymax.config) MODELED (waymax.config.ObjectType attribute) module waymax waymax.agents waymax.agents.actor_core waymax.agents.agent_builder waymax.agents.constant_speed waymax.agents.expert waymax.agents.sim_agent waymax.agents.waypoint_following_agent waymax.config waymax.dataloader waymax.dataloader.dataloader_utils waymax.dataloader.womd_dataloader waymax.dataloader.womd_factories waymax.dataloader.womd_factories_internal waymax.dataloader.womd_utils waymax.datatypes waymax.datatypes.action waymax.datatypes.array waymax.datatypes.constant waymax.datatypes.object_state waymax.datatypes.observation waymax.datatypes.operations waymax.datatypes.roadgraph waymax.datatypes.route waymax.datatypes.simulator_state waymax.datatypes.traffic_lights waymax.dynamics waymax.dynamics.abstract_dynamics waymax.dynamics.bicycle_model waymax.dynamics.delta waymax.dynamics.discretizer waymax.dynamics.state_dynamics waymax.env waymax.env.abstract_environment waymax.env.base_environment waymax.env.errors waymax.env.planning_agent_environment waymax.env.rollout waymax.env.typedefs waymax.env.wrappers waymax.env.wrappers.brax_wrapper waymax.env.wrappers.dm_env_wrapper waymax.metrics waymax.metrics.abstract_metric waymax.metrics.comfort waymax.metrics.imitation waymax.metrics.metric_factory waymax.metrics.overlap waymax.metrics.roadgraph waymax.metrics.route waymax.rewards waymax.rewards.abstract_reward_function waymax.rewards.linear_combination_reward waymax.utils waymax.utils.geometry waymax.utils.test_utils waymax.visualization waymax.visualization.color waymax.visualization.utils waymax.visualization.viz

MultiAgentEnvironment (in module waymax.env) (in module waymax.env.base_environment)

N

name (waymax.agents.actor_core.WaymaxActorCore property) (waymax.agents.sim_agent.SimAgentActor property) (waymax.agents.SimAgentActor property) (waymax.agents.WaymaxActorCore property) NON_SDC (waymax.config.ObjectType attribute) num_objects (waymax.datatypes.object_state.ObjectMetadata property) (waymax.datatypes.object_state.Trajectory property) (waymax.datatypes.ObjectMetadata property) (waymax.datatypes.ObjectPose2D property) (waymax.datatypes.Observation property) (waymax.datatypes.observation.ObjectPose2D property) (waymax.datatypes.observation.Observation property) (waymax.datatypes.simulator_state.SimulatorState property) (waymax.datatypes.SimulatorState property) (waymax.datatypes.Trajectory property)

num_paths (waymax.config.DatasetConfig attribute), [1] (waymax.datatypes.Paths property) (waymax.datatypes.route.Paths property) num_points (waymax.datatypes.roadgraph.RoadgraphPoints property) (waymax.datatypes.RoadgraphPoints property) num_points_per_path (waymax.config.DatasetConfig attribute), [1] (waymax.datatypes.Paths property) (waymax.datatypes.route.Paths property) num_shards (waymax.config.DatasetConfig attribute), [1] num_steps (waymax.config.ObservationConfig attribute), [1] num_timesteps (waymax.datatypes.object_state.Trajectory property) (waymax.datatypes.traffic_lights.TrafficLights property) (waymax.datatypes.TrafficLights property) (waymax.datatypes.Trajectory property) num_traffic_lights (waymax.datatypes.traffic_lights.TrafficLights property) (waymax.datatypes.TrafficLights property)

O

OBJECT (waymax.config.CoordinateFrame attribute) object_metadata (waymax.datatypes.simulator_state.SimulatorState attribute), [1] (waymax.datatypes.SimulatorState attribute), [1] object_metadata_from_womd_dict() (in module waymax.dataloader) (in module waymax.dataloader.womd_factories) object_types (waymax.datatypes.object_state.ObjectMetadata attribute), [1] (waymax.datatypes.ObjectMetadata attribute), [1] ObjectMetadata (class in waymax.datatypes) (class in waymax.datatypes.object_state) ObjectPose2D (class in waymax.datatypes) (class in waymax.datatypes.observation) objects_of_interest (waymax.datatypes.object_state.ObjectMetadata attribute), [1] (waymax.datatypes.ObjectMetadata attribute), [1] ObjectType (class in waymax.config) ObjectTypeIds (class in waymax.datatypes) (class in waymax.datatypes.object_state) Observation (class in waymax.datatypes) (class in waymax.datatypes.observation) (in module waymax.env) (in module waymax.env.typedefs) observation (waymax.config.EnvironmentConfig attribute), [1] (waymax.env.rollout.RolloutCarry attribute), [1] (waymax.env.rollout.RolloutOutput attribute), [1] (waymax.env.RolloutOutput attribute), [1] (waymax.env.wrappers.brax_wrapper.TimeStep attribute), [1] observation_from_state() (in module waymax.datatypes) (in module waymax.datatypes.observation) observation_spec() (waymax.env.abstract_environment.AbstractEnvironment method) (waymax.env.AbstractEnvironment method) (waymax.env.base_environment.BaseEnvironment method) (waymax.env.BaseEnvironment method) (waymax.env.BraxWrapper method) (waymax.env.DMEnvWrapper method) (waymax.env.planning_agent_environment.PlanningAgentEnvironment method) (waymax.env.PlanningAgentEnvironment method) (waymax.env.wrappers.brax_wrapper.BraxWrapper method) (waymax.env.wrappers.dm_env_wrapper.DMEnvWrapper method)

ObservationConfig (class in waymax.config) ObservationFn (in module waymax.env) (in module waymax.env.typedefs) observe() (waymax.env.abstract_environment.AbstractEnvironment method) (waymax.env.AbstractEnvironment method) (waymax.env.base_environment.BaseEnvironment method) (waymax.env.BaseEnvironment method) (waymax.env.BraxWrapper method) (waymax.env.DMEnvWrapper method) (waymax.env.planning_agent_environment.PlanningAgentEnvironment method) (waymax.env.PlanningAgentEnvironment method) (waymax.env.wrappers.brax_wrapper.BraxWrapper method) (waymax.env.wrappers.dm_env_wrapper.DMEnvWrapper method) OffroadMetric (class in waymax.metrics) (class in waymax.metrics.roadgraph) OffRouteMetric (class in waymax.metrics) (class in waymax.metrics.route) on_route (waymax.datatypes.Paths attribute), [1] (waymax.datatypes.route.Paths attribute), [1] original_xy (waymax.datatypes.ObjectPose2D attribute), [1] (waymax.datatypes.observation.ObjectPose2D attribute), [1] original_yaw (waymax.datatypes.ObjectPose2D attribute), [1] (waymax.datatypes.observation.ObjectPose2D attribute), [1] OTHER (waymax.datatypes.object_state.ObjectTypeIds attribute) (waymax.datatypes.ObjectTypeIds attribute) OverlapMetric (class in waymax.metrics) (class in waymax.metrics.overlap)

P

Params (in module waymax.agents.actor_core) path (waymax.config.DatasetConfig attribute), [1] Paths (class in waymax.datatypes) (class in waymax.datatypes.route) paths_from_womd_dict() (in module waymax.dataloader) (in module waymax.dataloader.womd_factories) PEDESTRIAN (waymax.datatypes.object_state.ObjectTypeIds attribute) (waymax.datatypes.ObjectTypeIds attribute) PlanningAgentDynamics (class in waymax.env) (class in waymax.env.planning_agent_environment) PlanningAgentEnvironment (class in waymax.env) (class in waymax.env.planning_agent_environment) PlanningAgentSimulatorState (class in waymax.env) (class in waymax.env.planning_agent_environment) plot_numpy_bounding_boxes() (in module waymax.visualization.utils) plot_observation() (in module waymax.visualization) (in module waymax.visualization.viz) plot_roadgraph_points() (in module waymax.visualization) (in module waymax.visualization.viz) plot_simulator_state() (in module waymax.visualization) (in module waymax.visualization.viz) plot_single_agent_brax_timestep() (in module waymax.visualization) (in module waymax.visualization.viz)

plot_traffic_light_signals_as_points() (in module waymax.visualization) (in module waymax.visualization.viz) plot_trajectory() (in module waymax.visualization) (in module waymax.visualization.viz) pose2d (waymax.datatypes.Observation attribute), [1] (waymax.datatypes.observation.Observation attribute), [1] pose_from_xy_yaw() (in module waymax.utils.geometry) preprocess_serialized_womd_data() (in module waymax.dataloader) (in module waymax.dataloader.womd_dataloader) preprocess_womd_example() (in module waymax.dataloader) (in module waymax.dataloader.womd_dataloader) ProgressionMetric (class in waymax.metrics) (class in waymax.metrics.route) px_per_meter (waymax.visualization.utils.VizConfig attribute) PyTree (in module waymax.datatypes) (in module waymax.datatypes.array) (in module waymax.datatypes.operations) (in module waymax.datatypes.roadgraph) (in module waymax.datatypes.simulator_state) (in module waymax.env) (in module waymax.env.typedefs)

R

register_actor_core() (in module waymax.agents.actor_core) register_metric() (in module waymax.metrics) (in module waymax.metrics.metric_factory) remaining_timesteps (waymax.datatypes.simulator_state.SimulatorState property) (waymax.datatypes.SimulatorState property) repeat (waymax.config.DatasetConfig attribute), [1] reset() (waymax.env.abstract_environment.AbstractEnvironment method) (waymax.env.AbstractEnvironment method) (waymax.env.base_environment.BaseEnvironment method) (waymax.env.BaseEnvironment method) (waymax.env.BraxWrapper method) (waymax.env.DMEnvWrapper method) (waymax.env.planning_agent_environment.PlanningAgentEnvironment method) (waymax.env.PlanningAgentEnvironment method) (waymax.env.wrappers.brax_wrapper.BraxWrapper method) (waymax.env.wrappers.dm_env_wrapper.DMEnvWrapper method) reward (waymax.env.rollout.RolloutOutput attribute), [1] (waymax.env.RolloutOutput attribute), [1] (waymax.env.wrappers.brax_wrapper.TimeStep attribute), [1] reward() (waymax.env.abstract_environment.AbstractEnvironment method) (waymax.env.AbstractEnvironment method) (waymax.env.base_environment.BaseEnvironment method) (waymax.env.BaseEnvironment method) (waymax.env.BraxWrapper method) (waymax.env.planning_agent_environment.PlanningAgentEnvironment method) (waymax.env.PlanningAgentEnvironment method) (waymax.env.wrappers.brax_wrapper.BraxWrapper method) reward_spec() (waymax.env.abstract_environment.AbstractEnvironment method) (waymax.env.AbstractEnvironment method) (waymax.env.base_environment.BaseEnvironment method) (waymax.env.BaseEnvironment method) (waymax.env.BraxWrapper method) (waymax.env.DMEnvWrapper method) (waymax.env.planning_agent_environment.PlanningAgentEnvironment method) (waymax.env.PlanningAgentEnvironment method) (waymax.env.wrappers.brax_wrapper.BraxWrapper method) (waymax.env.wrappers.dm_env_wrapper.DMEnvWrapper method) RewardFn (in module waymax.env) (in module waymax.env.typedefs) rewards (waymax.config.EnvironmentConfig attribute), [1] (waymax.config.LinearCombinationRewardConfig attribute), [1] rng (waymax.env.rollout.RolloutCarry attribute), [1] ROAD_EDGE_BOUNDARY (waymax.datatypes.MapElementIds attribute) (waymax.datatypes.roadgraph.MapElementIds attribute) ROAD_EDGE_MEDIAN (waymax.datatypes.MapElementIds attribute) (waymax.datatypes.roadgraph.MapElementIds attribute)

ROAD_EDGE_UNKNOWN (waymax.datatypes.MapElementIds attribute) (waymax.datatypes.roadgraph.MapElementIds attribute) ROAD_GRAPH_COLORS (in module waymax.visualization.color) ROAD_LINE_BROKEN_DOUBLE_YELLOW (waymax.datatypes.MapElementIds attribute) (waymax.datatypes.roadgraph.MapElementIds attribute) ROAD_LINE_BROKEN_SINGLE_WHITE (waymax.datatypes.MapElementIds attribute) (waymax.datatypes.roadgraph.MapElementIds attribute) ROAD_LINE_BROKEN_SINGLE_YELLOW (waymax.datatypes.MapElementIds attribute) (waymax.datatypes.roadgraph.MapElementIds attribute) ROAD_LINE_PASSING_DOUBLE_YELLOW (waymax.datatypes.MapElementIds attribute) (waymax.datatypes.roadgraph.MapElementIds attribute) ROAD_LINE_SOLID_DOUBLE_WHITE (waymax.datatypes.MapElementIds attribute) (waymax.datatypes.roadgraph.MapElementIds attribute) ROAD_LINE_SOLID_DOUBLE_YELLOW (waymax.datatypes.MapElementIds attribute) (waymax.datatypes.roadgraph.MapElementIds attribute) ROAD_LINE_SOLID_SINGLE_WHITE (waymax.datatypes.MapElementIds attribute) (waymax.datatypes.roadgraph.MapElementIds attribute) ROAD_LINE_SOLID_SINGLE_YELLOW (waymax.datatypes.MapElementIds attribute) (waymax.datatypes.roadgraph.MapElementIds attribute) ROAD_LINE_UNKNOWN (waymax.datatypes.MapElementIds attribute) (waymax.datatypes.roadgraph.MapElementIds attribute) roadgraph_from_womd_dict() (in module waymax.dataloader) (in module waymax.dataloader.womd_factories) roadgraph_points (waymax.datatypes.simulator_state.SimulatorState attribute), [1] (waymax.datatypes.SimulatorState attribute), [1] roadgraph_static_points (waymax.datatypes.Observation attribute), [1] (waymax.datatypes.observation.Observation attribute), [1] roadgraph_top_k (waymax.config.ObservationConfig attribute), [1] RoadgraphPoints (class in waymax.datatypes) (class in waymax.datatypes.roadgraph) rollout() (in module waymax.env) (in module waymax.env.rollout) rollout_log_by_expert_sdc() (in module waymax.env) (in module waymax.env.rollout) RolloutCarry (class in waymax.env.rollout) RolloutOutput (class in waymax.env) (class in waymax.env.rollout) rotate_points() (in module waymax.utils.geometry) rotation_matrix (waymax.datatypes.ObjectPose2D property) (waymax.datatypes.observation.ObjectPose2D property) rotation_matrix_2d() (in module waymax.utils.geometry) ROUTE_DATA_PATH (in module waymax.utils.test_utils) ROUTE_NUM_PATHS (in module waymax.utils.test_utils) ROUTE_NUM_POINTS_PER_PATH (in module waymax.utils.test_utils) run_metrics() (in module waymax.metrics) (in module waymax.metrics.metric_factory)

S

save_img_as_png() (in module waymax.visualization.utils) SDC (waymax.config.CoordinateFrame attribute) (waymax.config.ObjectType attribute) sdc_observation_from_state() (in module waymax.datatypes) (in module waymax.datatypes.observation) sdc_paths (waymax.datatypes.Observation attribute), [1] (waymax.datatypes.observation.Observation attribute), [1] (waymax.datatypes.simulator_state.SimulatorState attribute), [1] (waymax.datatypes.SimulatorState attribute), [1] select_action() (waymax.agents.actor_core.WaymaxActorCore method) (waymax.agents.sim_agent.SimAgentActor method) (waymax.agents.SimAgentActor method) (waymax.agents.WaymaxActorCore method) select_by_onehot() (in module waymax.datatypes) (in module waymax.datatypes.operations) shape (waymax.datatypes.Action property) (waymax.datatypes.action.Action property) (waymax.datatypes.action.TrajectoryUpdate property) (waymax.datatypes.array.MaskedArray property) (waymax.datatypes.MaskedArray property) (waymax.datatypes.object_state.ObjectMetadata property) (waymax.datatypes.object_state.Trajectory property) (waymax.datatypes.ObjectMetadata property) (waymax.datatypes.ObjectPose2D property) (waymax.datatypes.Observation property) (waymax.datatypes.observation.ObjectPose2D property) (waymax.datatypes.observation.Observation property) (waymax.datatypes.Paths property) (waymax.datatypes.roadgraph.RoadgraphPoints property) (waymax.datatypes.RoadgraphPoints property) (waymax.datatypes.route.Paths property) (waymax.datatypes.simulator_state.SimulatorState property) (waymax.datatypes.SimulatorState property) (waymax.datatypes.traffic_lights.TrafficLights property) (waymax.datatypes.TrafficLights property) (waymax.datatypes.Trajectory property) (waymax.datatypes.TrajectoryUpdate property) (waymax.env.rollout.RolloutOutput property) (waymax.env.RolloutOutput property) (waymax.env.wrappers.brax_wrapper.TimeStep property) show_agent_id (waymax.visualization.utils.VizConfig attribute) shuffle_buffer_size (waymax.config.DatasetConfig attribute), [1] shuffle_seed (waymax.config.DatasetConfig attribute), [1] sim_agent_actor_states (waymax.env.planning_agent_environment.PlanningAgentSimulatorState attribute), [1] (waymax.env.PlanningAgentSimulatorState attribute), [1] sim_agents (waymax.config.EnvironmentConfig attribute), [1] sim_trajectory (waymax.datatypes.simulator_state.SimulatorState attribute), [1] (waymax.datatypes.SimulatorState attribute), [1] SimAgentActor (class in waymax.agents) (class in waymax.agents.sim_agent)

SimAgentConfig (class in waymax.config) SimAgentType (class in waymax.config) simulated_trajectory_no_overlap() (in module waymax.utils.test_utils) simulated_trajectory_with_overlap() (in module waymax.utils.test_utils) simulation_state (waymax.env.DMEnvWrapper property) (waymax.env.wrappers.dm_env_wrapper.DMEnvWrapper property) SimulationNotInitializedError, [1] simulator_state_from_womd_dict() (in module waymax.dataloader) (in module waymax.dataloader.womd_factories) simulator_state_generator() (in module waymax.dataloader) (in module waymax.dataloader.womd_dataloader) simulator_state_to_womd_dict() (in module waymax.dataloader) (in module waymax.dataloader.womd_utils) simulator_state_to_womd_dict_tensorflow() (in module waymax.dataloader.womd_utils) simulator_state_with_offroad() (in module waymax.utils.test_utils) simulator_state_with_overlap() (in module waymax.utils.test_utils) simulator_state_without_offroad() (in module waymax.utils.test_utils) SimulatorState (class in waymax.datatypes) (class in waymax.datatypes.simulator_state) speed (waymax.datatypes.object_state.Trajectory property) (waymax.datatypes.Trajectory property) SPEED_BUMP (waymax.datatypes.MapElementIds attribute) (waymax.datatypes.roadgraph.MapElementIds attribute) stack_fields() (waymax.datatypes.object_state.Trajectory method) (waymax.datatypes.Trajectory method) state (waymax.datatypes.traffic_lights.TrafficLights attribute), [1] (waymax.datatypes.TrafficLights attribute), [1] (waymax.env.rollout.RolloutCarry attribute), [1] (waymax.env.rollout.RolloutOutput attribute), [1] (waymax.env.RolloutOutput attribute), [1] (waymax.env.wrappers.brax_wrapper.TimeStep attribute), [1] StateDynamics (class in waymax.dynamics) (class in waymax.dynamics.state_dynamics) stateless_env (waymax.env.DMEnvWrapper property) (waymax.env.wrappers.dm_env_wrapper.DMEnvWrapper property) step() (waymax.env.abstract_environment.AbstractEnvironment method) (waymax.env.AbstractEnvironment method) (waymax.env.base_environment.BaseEnvironment method) (waymax.env.BaseEnvironment method) (waymax.env.BraxWrapper method) (waymax.env.DMEnvWrapper method) (waymax.env.planning_agent_environment.PlanningAgentEnvironment method) (waymax.env.PlanningAgentEnvironment method) (waymax.env.wrappers.brax_wrapper.BraxWrapper method) (waymax.env.wrappers.dm_env_wrapper.DMEnvWrapper method) STOP (waymax.datatypes.traffic_lights.TrafficLightStates attribute) (waymax.datatypes.TrafficLightStates attribute) STOP_SIGN (waymax.datatypes.MapElementIds attribute) (waymax.datatypes.roadgraph.MapElementIds attribute)

T

T (in module waymax.dataloader.dataloader_utils) (in module waymax.dataloader.womd_dataloader) termination() (waymax.env.abstract_environment.AbstractEnvironment method) (waymax.env.AbstractEnvironment method) (waymax.env.BraxWrapper method) (waymax.env.wrappers.brax_wrapper.BraxWrapper method) tf_data_service_address (waymax.config.DatasetConfig attribute), [1] tf_examples_dataset() (in module waymax.dataloader) (in module waymax.dataloader.dataloader_utils) TFRECORD (waymax.config.DataFormat attribute) TIME_INTERVAL (in module waymax.datatypes) (in module waymax.datatypes.constant) timestamp_micros (waymax.datatypes.object_state.Trajectory attribute), [1] (waymax.datatypes.Trajectory attribute), [1] TimeStep (class in waymax.env.wrappers.brax_wrapper) timestep (waymax.datatypes.simulator_state.SimulatorState attribute), [1] (waymax.datatypes.SimulatorState attribute), [1] TIMESTEP_MICROS_INTERVAL (in module waymax.datatypes) (in module waymax.datatypes.constant) TL_STEP_AXIS (in module waymax.dataloader.womd_utils) TL_TIMESTAMP_STEP_AXIS (in module waymax.dataloader.womd_utils) TRAFFIC_LIGHT_COLORS (in module waymax.visualization.color) traffic_lights (waymax.datatypes.Observation attribute), [1] (waymax.datatypes.observation.Observation attribute), [1] traffic_lights_from_womd_dict() (in module waymax.dataloader) (in module waymax.dataloader.womd_factories) TrafficLights (class in waymax.datatypes) (class in waymax.datatypes.traffic_lights) TrafficLightStates (class in waymax.datatypes) (class in waymax.datatypes.traffic_lights)

Trajectory (class in waymax.datatypes) (class in waymax.datatypes.object_state) trajectory (waymax.datatypes.Observation attribute), [1] (waymax.datatypes.observation.Observation attribute), [1] trajectory_from_womd_dict() (in module waymax.dataloader) (in module waymax.dataloader.womd_factories) TrajectoryUpdate (class in waymax.datatypes) (class in waymax.datatypes.action) transform_direction() (in module waymax.utils.geometry) transform_observation() (in module waymax.datatypes) (in module waymax.datatypes.observation) transform_points() (in module waymax.utils.geometry) transform_roadgraph_points() (in module waymax.datatypes) (in module waymax.datatypes.observation) transform_traffic_lights() (in module waymax.datatypes) (in module waymax.datatypes.observation) transform_trajectories() (in module waymax.utils.geometry) transform_trajectory() (in module waymax.datatypes) (in module waymax.datatypes.observation) transform_yaw() (in module waymax.utils.geometry) translation_vector (waymax.datatypes.ObjectPose2D property) (waymax.datatypes.observation.ObjectPose2D property) truncation() (waymax.env.abstract_environment.AbstractEnvironment method) (waymax.env.AbstractEnvironment method) (waymax.env.BraxWrapper method) (waymax.env.wrappers.brax_wrapper.BraxWrapper method) types (waymax.datatypes.roadgraph.RoadgraphPoints attribute), [1] (waymax.datatypes.RoadgraphPoints attribute), [1]

U

UNKNOWN (waymax.datatypes.MapElementIds attribute) (waymax.datatypes.roadgraph.MapElementIds attribute) (waymax.datatypes.traffic_lights.TrafficLightStates attribute) (waymax.datatypes.TrafficLightStates attribute) UNSET (waymax.datatypes.object_state.ObjectTypeIds attribute) (waymax.datatypes.ObjectTypeIds attribute) update_by_mask() (in module waymax.datatypes) (in module waymax.datatypes.operations) update_by_slice_in_dim() (in module waymax.datatypes) (in module waymax.datatypes.operations) update_speed() (waymax.agents.constant_speed.ConstantSpeedPolicy method) (waymax.agents.IDMRoutePolicy method) (waymax.agents.waypoint_following_agent.IDMRoutePolicy method) (waymax.agents.waypoint_following_agent.WaypointFollowingPolicy method) (waymax.agents.WaypointFollowingPolicy method)

update_state_by_log() (in module waymax.datatypes) (in module waymax.datatypes.simulator_state) update_trajectory() (waymax.agents.FrozenSimPolicy method) (waymax.agents.sim_agent.FrozenSimPolicy method) (waymax.agents.sim_agent.SimAgentActor method) (waymax.agents.SimAgentActor method) (waymax.agents.waypoint_following_agent.WaypointFollowingPolicy method) (waymax.agents.WaypointFollowingPolicy method)

V

VALID (waymax.config.ObjectType attribute) valid (waymax.datatypes.Action attribute), [1] (waymax.datatypes.action.Action attribute), [1] (waymax.datatypes.action.TrajectoryUpdate attribute) (waymax.datatypes.array.MaskedArray attribute), [1] (waymax.datatypes.MaskedArray attribute), [1] (waymax.datatypes.object_state.Trajectory attribute), [1] (waymax.datatypes.ObjectPose2D attribute), [1] (waymax.datatypes.Observation property) (waymax.datatypes.observation.ObjectPose2D attribute), [1] (waymax.datatypes.observation.Observation property) (waymax.datatypes.Paths attribute), [1] (waymax.datatypes.roadgraph.RoadgraphPoints attribute), [1] (waymax.datatypes.RoadgraphPoints attribute), [1] (waymax.datatypes.route.Paths attribute), [1] (waymax.datatypes.traffic_lights.TrafficLights attribute), [1] (waymax.datatypes.TrafficLights attribute), [1] (waymax.datatypes.Trajectory attribute), [1] (waymax.datatypes.TrajectoryUpdate attribute) (waymax.metrics.abstract_metric.MetricResult attribute) (waymax.metrics.MetricResult attribute) validate() (waymax.agents.actor_core.WaymaxActorOutput method) (waymax.agents.WaymaxActorOutput method) (waymax.datatypes.Action method) (waymax.datatypes.action.Action method) (waymax.datatypes.action.TrajectoryUpdate method) (waymax.datatypes.array.MaskedArray method) (waymax.datatypes.MaskedArray method) (waymax.datatypes.object_state.ObjectMetadata method) (waymax.datatypes.object_state.Trajectory method) (waymax.datatypes.ObjectMetadata method) (waymax.datatypes.ObjectPose2D method) (waymax.datatypes.Observation method) (waymax.datatypes.observation.ObjectPose2D method) (waymax.datatypes.observation.Observation method) (waymax.datatypes.Paths method) (waymax.datatypes.roadgraph.RoadgraphPoints method) (waymax.datatypes.RoadgraphPoints method) (waymax.datatypes.route.Paths method) (waymax.datatypes.simulator_state.SimulatorState method) (waymax.datatypes.SimulatorState method) (waymax.datatypes.traffic_lights.TrafficLights method) (waymax.datatypes.TrafficLights method) (waymax.datatypes.Trajectory method) (waymax.datatypes.TrajectoryUpdate method) (waymax.env.rollout.RolloutOutput method) (waymax.env.RolloutOutput method) (waymax.metrics.abstract_metric.MetricResult method) (waymax.metrics.MetricResult method)

value (waymax.datatypes.array.MaskedArray attribute), [1] (waymax.datatypes.MaskedArray attribute), [1] (waymax.metrics.abstract_metric.MetricResult attribute) (waymax.metrics.MetricResult attribute) VEHICLE (waymax.datatypes.object_state.ObjectTypeIds attribute) (waymax.datatypes.ObjectTypeIds attribute) vel_x (waymax.datatypes.action.TrajectoryUpdate attribute) (waymax.datatypes.object_state.Trajectory attribute), [1] (waymax.datatypes.Trajectory attribute), [1] (waymax.datatypes.TrajectoryUpdate attribute) vel_xy (waymax.datatypes.object_state.Trajectory property) (waymax.datatypes.Trajectory property) vel_y (waymax.datatypes.action.TrajectoryUpdate attribute) (waymax.datatypes.object_state.Trajectory attribute), [1] (waymax.datatypes.Trajectory attribute), [1] (waymax.datatypes.TrajectoryUpdate attribute) vel_yaw (waymax.datatypes.object_state.Trajectory property) (waymax.datatypes.Trajectory property) verbose (waymax.visualization.utils.VizConfig attribute) VizConfig (class in waymax.visualization.utils)

W

waymax module waymax.agents module waymax.agents.actor_core module waymax.agents.agent_builder module waymax.agents.constant_speed module waymax.agents.expert module waymax.agents.sim_agent module waymax.agents.waypoint_following_agent module waymax.config module waymax.dataloader module waymax.dataloader.dataloader_utils module waymax.dataloader.womd_dataloader module waymax.dataloader.womd_factories module waymax.dataloader.womd_factories_internal module waymax.dataloader.womd_utils module waymax.datatypes module waymax.datatypes.action module waymax.datatypes.array module waymax.datatypes.constant module waymax.datatypes.object_state module waymax.datatypes.observation module waymax.datatypes.operations module waymax.datatypes.roadgraph module waymax.datatypes.route module waymax.datatypes.simulator_state module waymax.datatypes.traffic_lights module waymax.dynamics module waymax.dynamics.abstract_dynamics module waymax.dynamics.bicycle_model module waymax.dynamics.delta module waymax.dynamics.discretizer module waymax.dynamics.state_dynamics module waymax.env module waymax.env.abstract_environment module waymax.env.base_environment module

waymax.env.errors module waymax.env.planning_agent_environment module waymax.env.rollout module waymax.env.typedefs module waymax.env.wrappers module waymax.env.wrappers.brax_wrapper module waymax.env.wrappers.dm_env_wrapper module waymax.metrics module waymax.metrics.abstract_metric module waymax.metrics.comfort module waymax.metrics.imitation module waymax.metrics.metric_factory module waymax.metrics.overlap module waymax.metrics.roadgraph module waymax.metrics.route module waymax.rewards module waymax.rewards.abstract_reward_function module waymax.rewards.linear_combination_reward module waymax.utils module waymax.utils.geometry module waymax.utils.test_utils module waymax.visualization module waymax.visualization.color module waymax.visualization.utils module waymax.visualization.viz module WaymaxActorCore (class in waymax.agents) (class in waymax.agents.actor_core) WaymaxActorOutput (class in waymax.agents) (class in waymax.agents.actor_core) WaymaxConfig (class in waymax.config) WaypointFollowingPolicy (class in waymax.agents) (class in waymax.agents.waypoint_following_agent) width (waymax.datatypes.object_state.Trajectory attribute), [1] (waymax.datatypes.Trajectory attribute), [1] WOD_1_0_0_TRAINING (in module waymax.config) WOD_1_0_0_VALIDATION (in module waymax.config) WOD_1_1_0_TRAINING (in module waymax.config) WOD_1_1_0_VALIDATION (in module waymax.config) WOD_1_2_0_TEST (in module waymax.config) wrap_yaws() (in module waymax.utils.geometry) WRONG_WAY_THRES (waymax.metrics.roadgraph.WrongWayMetric attribute) (waymax.metrics.WrongWayMetric attribute) WrongWayMetric (class in waymax.metrics) (class in waymax.metrics.roadgraph)

X

x (waymax.datatypes.action.TrajectoryUpdate attribute) (waymax.datatypes.object_state.Trajectory attribute), [1] (waymax.datatypes.Paths attribute), [1] (waymax.datatypes.roadgraph.RoadgraphPoints attribute), [1] (waymax.datatypes.RoadgraphPoints attribute), [1] (waymax.datatypes.route.Paths attribute), [1] (waymax.datatypes.traffic_lights.TrafficLights attribute), [1] (waymax.datatypes.TrafficLights attribute), [1] (waymax.datatypes.Trajectory attribute), [1] (waymax.datatypes.TrajectoryUpdate attribute) xy (waymax.datatypes.object_state.Trajectory property) (waymax.datatypes.Paths property) (waymax.datatypes.roadgraph.RoadgraphPoints property) (waymax.datatypes.RoadgraphPoints property) (waymax.datatypes.route.Paths property) (waymax.datatypes.traffic_lights.TrafficLights property) (waymax.datatypes.TrafficLights property) (waymax.datatypes.Trajectory property)

xyz (waymax.datatypes.object_state.Trajectory property) (waymax.datatypes.Paths property) (waymax.datatypes.roadgraph.RoadgraphPoints property) (waymax.datatypes.RoadgraphPoints property) (waymax.datatypes.route.Paths property) (waymax.datatypes.Trajectory property)

Y

y (waymax.datatypes.action.TrajectoryUpdate attribute) (waymax.datatypes.object_state.Trajectory attribute), [1] (waymax.datatypes.Paths attribute), [1] (waymax.datatypes.roadgraph.RoadgraphPoints attribute), [1] (waymax.datatypes.RoadgraphPoints attribute), [1] (waymax.datatypes.route.Paths attribute), [1] (waymax.datatypes.traffic_lights.TrafficLights attribute), [1] (waymax.datatypes.TrafficLights attribute), [1] (waymax.datatypes.Trajectory attribute), [1] (waymax.datatypes.TrajectoryUpdate attribute)

yaw (waymax.datatypes.action.TrajectoryUpdate attribute) (waymax.datatypes.object_state.Trajectory attribute), [1] (waymax.datatypes.Trajectory attribute), [1] (waymax.datatypes.TrajectoryUpdate attribute)

Z

z (waymax.datatypes.object_state.Trajectory attribute), [1] (waymax.datatypes.Paths attribute), [1] (waymax.datatypes.roadgraph.RoadgraphPoints attribute), [1] (waymax.datatypes.RoadgraphPoints attribute), [1] (waymax.datatypes.route.Paths attribute), [1] (waymax.datatypes.traffic_lights.TrafficLights attribute), [1] (waymax.datatypes.TrafficLights attribute), [1] (waymax.datatypes.Trajectory attribute), [1]

zeros() (waymax.datatypes.object_state.Trajectory class method) (waymax.datatypes.Trajectory class method)