JAXAtari
API
Environment Base
Core
Wrappers
Spaces
Rendering
Game Environments
Scripts
RAMStateDeltas
Frame Extractor
Sprite Editor
Tests & Benchmarks
Benchmarks
JAXAtari
Index
Index
A
|
B
|
C
|
D
|
E
|
F
|
G
|
H
|
I
|
J
|
K
|
L
|
M
|
N
|
O
|
P
|
R
|
S
|
T
|
U
|
V
|
W
|
X
|
Y
A
acceleration_counter (jaxatari.games.jax_pong.PongState attribute)
ACTION_SET (jaxatari.games.jax_freeway.JaxFreeway attribute)
(jaxatari.games.jax_kangaroo.JaxKangaroo attribute)
(jaxatari.games.jax_pong.JaxPong attribute)
(jaxatari.games.jax_seaquest.JaxSeaquest attribute)
action_space() (jaxatari.environment.JaxEnvironment method)
(jaxatari.games.jax_freeway.JaxFreeway method)
(jaxatari.games.jax_kangaroo.JaxKangaroo method)
(jaxatari.games.jax_pong.JaxPong method)
(jaxatari.games.jax_seaquest.JaxSeaquest method)
active (jaxatari.games.jax_seaquest.EntityPosition attribute)
(jaxatari.games.jax_seaquest.PlayerEntity attribute)
ASSET_CONFIG (jaxatari.games.jax_freeway.FreewayConstants attribute)
(jaxatari.games.jax_kangaroo.KangarooConstants attribute)
(jaxatari.games.jax_pong.PongConstants attribute)
(jaxatari.games.jax_seaquest.SeaquestConstants attribute)
atari_state (jaxatari.wrappers.LogState attribute)
(jaxatari.wrappers.MultiRewardLogState attribute)
(jaxatari.wrappers.PixelAndObjectCentricState attribute)
(jaxatari.wrappers.PixelState attribute)
AtariState (class in jaxatari.wrappers)
AtariWrapper (class in jaxatari.wrappers)
B
BACKGROUND_COLOR (jaxatari.games.jax_pong.PongConstants attribute)
(jaxatari.games.jax_seaquest.SeaquestConstants attribute)
ball (jaxatari.games.jax_pong.PongObservation attribute)
BALL_COLOR (jaxatari.games.jax_pong.PongConstants attribute)
BALL_MAX_SPEED (jaxatari.games.jax_pong.PongConstants attribute)
BALL_SIZE (jaxatari.games.jax_pong.PongConstants attribute)
BALL_SPEED (jaxatari.games.jax_pong.PongConstants attribute)
BALL_START_X (jaxatari.games.jax_pong.PongConstants attribute)
BALL_START_Y (jaxatari.games.jax_pong.PongConstants attribute)
ball_vel_x (jaxatari.games.jax_pong.PongState attribute)
ball_vel_y (jaxatari.games.jax_pong.PongState attribute)
ball_x (jaxatari.games.jax_pong.PongState attribute)
ball_y (jaxatari.games.jax_pong.PongState attribute)
BASE_BALL_SPEED (jaxatari.games.jax_pong.PongConstants attribute)
bell_animation (jaxatari.games.jax_kangaroo.LevelState attribute)
BELL_COLOR (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
BELL_HEIGHT (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
bell_position (jaxatari.games.jax_kangaroo.KangarooObservation attribute)
(jaxatari.games.jax_kangaroo.LevelState attribute)
bell_timer (jaxatari.games.jax_kangaroo.LevelState attribute)
BELL_WIDTH (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
blink_start_frames (jaxatari.games.jax_freeway.FreewayConstants attribute)
bottom_border (jaxatari.games.jax_freeway.FreewayConstants attribute)
Box (class in jaxatari.spaces)
buffer (jaxatari.games.jax_pong.PongState attribute)
C
cadence_phase_offset (jaxatari.games.jax_freeway.FreewayConstants attribute)
calculate_kill_points() (jaxatari.games.jax_seaquest.JaxSeaquest method)
car (jaxatari.games.jax_freeway.FreewayObservation attribute)
CAR_COLORS (jaxatari.games.jax_freeway.FreewayConstants attribute)
car_height (jaxatari.games.jax_freeway.FreewayConstants attribute)
car_hit_inset_x (jaxatari.games.jax_freeway.FreewayConstants attribute)
car_hit_inset_y_bottom (jaxatari.games.jax_freeway.FreewayConstants attribute)
car_hit_inset_y_top (jaxatari.games.jax_freeway.FreewayConstants attribute)
car_speeds (jaxatari.games.jax_freeway.FreewayConstants attribute)
CAR_UPDATES (jaxatari.games.jax_freeway.FreewayConstants attribute)
car_width (jaxatari.games.jax_freeway.FreewayConstants attribute)
car_y_offset (jaxatari.games.jax_freeway.FreewayConstants attribute)
CarryState (class in jaxatari.games.jax_seaquest)
cars (jaxatari.games.jax_freeway.FreewayState attribute)
check_collision_batch() (jaxatari.games.jax_seaquest.JaxSeaquest method)
check_collision_single() (jaxatari.games.jax_seaquest.JaxSeaquest method)
check_missile_collisions() (jaxatari.games.jax_seaquest.JaxSeaquest method)
check_player_collision() (jaxatari.games.jax_seaquest.JaxSeaquest method)
chicken (jaxatari.games.jax_freeway.FreewayObservation attribute)
chicken_height (jaxatari.games.jax_freeway.FreewayConstants attribute)
chicken_hit_inset_x (jaxatari.games.jax_freeway.FreewayConstants attribute)
chicken_hit_inset_y_bottom (jaxatari.games.jax_freeway.FreewayConstants attribute)
chicken_hit_inset_y_top (jaxatari.games.jax_freeway.FreewayConstants attribute)
chicken_width (jaxatari.games.jax_freeway.FreewayConstants attribute)
chicken_x (jaxatari.games.jax_freeway.FreewayConstants attribute)
chicken_y (jaxatari.games.jax_freeway.FreewayState attribute)
CHILD_HEIGHT (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
child_position (jaxatari.games.jax_kangaroo.KangarooObservation attribute)
(jaxatari.games.jax_kangaroo.LevelState attribute)
child_timer (jaxatari.games.jax_kangaroo.LevelState attribute)
child_velocity (jaxatari.games.jax_kangaroo.LevelState attribute)
CHILD_WIDTH (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
chrash_timer (jaxatari.games.jax_kangaroo.PlayerState attribute)
climb_base_y (jaxatari.games.jax_kangaroo.PlayerState attribute)
climb_counter (jaxatari.games.jax_kangaroo.PlayerState attribute)
coco_positions (jaxatari.games.jax_kangaroo.KangarooObservation attribute)
(jaxatari.games.jax_kangaroo.LevelState attribute)
coco_states (jaxatari.games.jax_kangaroo.LevelState attribute)
COCONUT_COLOR (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
collected_divers (jaxatari.games.jax_seaquest.SeaquestObservation attribute)
contains() (jaxatari.spaces.Box method)
(jaxatari.spaces.Dict method)
(jaxatari.spaces.Discrete method)
(jaxatari.spaces.Space method)
(jaxatari.spaces.Tuple method)
cooldown (jaxatari.games.jax_freeway.FreewayState attribute)
cooldown_counter (jaxatari.games.jax_kangaroo.PlayerState attribute)
current_level (jaxatari.games.jax_kangaroo.KangarooState attribute)
D
death_counter (jaxatari.games.jax_seaquest.SeaquestState attribute)
Dict (class in jaxatari.spaces)
difficulty (jaxatari.games.jax_seaquest.SeaquestInfo attribute)
(jaxatari.games.jax_seaquest.SpawnState attribute)
Discrete (class in jaxatari.spaces)
diver_array (jaxatari.games.jax_seaquest.SpawnState attribute)
DIVER_COLOR (jaxatari.games.jax_seaquest.SeaquestConstants attribute)
diver_positions (jaxatari.games.jax_seaquest.SeaquestState attribute)
DIVER_SIZE (jaxatari.games.jax_seaquest.SeaquestConstants attribute)
DIVER_SPAWN_POSITIONS (jaxatari.games.jax_seaquest.SeaquestConstants attribute)
divers (jaxatari.games.jax_seaquest.SeaquestObservation attribute)
divers_collected (jaxatari.games.jax_seaquest.SeaquestState attribute)
DOWN (jaxatari.environment.JAXAtariAction attribute)
DOWNFIRE (jaxatari.environment.JAXAtariAction attribute)
DOWNLEFT (jaxatari.environment.JAXAtariAction attribute)
DOWNLEFTFIRE (jaxatari.environment.JAXAtariAction attribute)
DOWNRIGHT (jaxatari.environment.JAXAtariAction attribute)
DOWNRIGHTFIRE (jaxatari.environment.JAXAtariAction attribute)
E
enemy (jaxatari.games.jax_pong.PongObservation attribute)
ENEMY_COLOR (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
(jaxatari.games.jax_pong.PongConstants attribute)
ENEMY_HEIGHT (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
enemy_missile_positions (jaxatari.games.jax_seaquest.SeaquestState attribute)
ENEMY_MISSILE_Y (jaxatari.games.jax_seaquest.SeaquestConstants attribute)
enemy_missiles (jaxatari.games.jax_seaquest.SeaquestObservation attribute)
enemy_missiles_step() (jaxatari.games.jax_seaquest.JaxSeaquest method)
enemy_score (jaxatari.games.jax_pong.PongState attribute)
ENEMY_SIZE (jaxatari.games.jax_pong.PongConstants attribute)
enemy_speed (jaxatari.games.jax_pong.PongState attribute)
ENEMY_STEP_SIZE (jaxatari.games.jax_pong.PongConstants attribute)
ENEMY_SUB_COLOR (jaxatari.games.jax_seaquest.SeaquestConstants attribute)
ENEMY_SUB_SIZE (jaxatari.games.jax_seaquest.SeaquestConstants attribute)
ENEMY_WIDTH (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
ENEMY_X (jaxatari.games.jax_pong.PongConstants attribute)
enemy_y (jaxatari.games.jax_pong.PongState attribute)
Entity (class in jaxatari.games.jax_kangaroo)
EntityPosition (class in jaxatari.games.jax_freeway)
(class in jaxatari.games.jax_pong)
(class in jaxatari.games.jax_seaquest)
env_state (jaxatari.wrappers.AtariState attribute)
episode_lengths (jaxatari.wrappers.LogState attribute)
(jaxatari.wrappers.MultiRewardLogState attribute)
episode_returns (jaxatari.wrappers.LogState attribute)
(jaxatari.wrappers.MultiRewardLogState attribute)
episode_returns_env (jaxatari.wrappers.MultiRewardLogState attribute)
F
FACE_LEFT (jaxatari.games.jax_seaquest.SeaquestConstants attribute)
FACE_RIGHT (jaxatari.games.jax_seaquest.SeaquestConstants attribute)
falling_coco_counter (jaxatari.games.jax_kangaroo.LevelState attribute)
falling_coco_dropping (jaxatari.games.jax_kangaroo.LevelState attribute)
falling_coco_position (jaxatari.games.jax_kangaroo.KangarooObservation attribute)
(jaxatari.games.jax_kangaroo.LevelState attribute)
falling_coco_skip_update (jaxatari.games.jax_kangaroo.LevelState attribute)
FALLING_COCONUT_HEIGHT (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
FALLING_COCONUT_WIDTH (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
FIRE (jaxatari.environment.JAXAtariAction attribute)
FIRST_WAVE_DIRS (jaxatari.games.jax_seaquest.SeaquestConstants attribute)
flatten_entity_position() (jaxatari.games.jax_seaquest.JaxSeaquest method)
flatten_player_entity() (jaxatari.games.jax_seaquest.JaxSeaquest method)
FlattenObservationWrapper (class in jaxatari.wrappers)
FreewayConstants (class in jaxatari.games.jax_freeway)
FreewayInfo (class in jaxatari.games.jax_freeway)
FreewayObservation (class in jaxatari.games.jax_freeway)
FreewayRenderer (class in jaxatari.games.jax_freeway)
FreewayState (class in jaxatari.games.jax_freeway)
fruit_actives (jaxatari.games.jax_kangaroo.LevelState attribute)
FRUIT_COLOR (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
FRUIT_COLOR_STATE_1 (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
FRUIT_COLOR_STATE_2 (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
FRUIT_COLOR_STATE_3 (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
FRUIT_COLOR_STATE_4 (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
FRUIT_HEIGHT (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
fruit_positions (jaxatari.games.jax_kangaroo.KangarooObservation attribute)
(jaxatari.games.jax_kangaroo.LevelState attribute)
fruit_stages (jaxatari.games.jax_kangaroo.LevelState attribute)
FRUIT_WIDTH (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
G
game_duration_frames (jaxatari.games.jax_freeway.FreewayConstants attribute)
game_over (jaxatari.games.jax_freeway.FreewayState attribute)
get_all_values() (jaxatari.environment.JAXAtariAction class method)
get_default_asset_config() (in module jaxatari.games.jax_kangaroo)
get_front_entity() (jaxatari.games.jax_seaquest.JaxSeaquest method)
get_pattern_for_difficulty() (jaxatari.games.jax_seaquest.JaxSeaquest method)
get_shark_color_index() (in module jaxatari.games.jax_seaquest)
get_spawn_position() (jaxatari.games.jax_seaquest.JaxSeaquest method)
H
h (jaxatari.games.jax_kangaroo.Entity attribute)
height (jaxatari.games.jax_freeway.EntityPosition attribute)
(jaxatari.games.jax_kangaroo.PlayerState attribute)
(jaxatari.games.jax_pong.EntityPosition attribute)
HEIGHT (jaxatari.games.jax_pong.PongConstants attribute)
height (jaxatari.games.jax_seaquest.EntityPosition attribute)
(jaxatari.games.jax_seaquest.PlayerEntity attribute)
I
image_space() (jaxatari.environment.JaxEnvironment method)
(jaxatari.games.jax_freeway.JaxFreeway method)
(jaxatari.games.jax_kangaroo.JaxKangaroo method)
(jaxatari.games.jax_pong.JaxPong method)
(jaxatari.games.jax_seaquest.JaxSeaquest method)
(jaxatari.wrappers.AtariWrapper method)
image_stack (jaxatari.wrappers.PixelAndObjectCentricState attribute)
(jaxatari.wrappers.PixelState attribute)
initialize_spawn_state() (jaxatari.games.jax_seaquest.JaxSeaquest method)
is_climbing (jaxatari.games.jax_kangaroo.PlayerState attribute)
is_crashing (jaxatari.games.jax_kangaroo.PlayerState attribute)
is_crouching (jaxatari.games.jax_kangaroo.PlayerState attribute)
is_jumping (jaxatari.games.jax_kangaroo.PlayerState attribute)
is_slot_empty() (jaxatari.games.jax_seaquest.JaxSeaquest method)
J
jaxatari.core
module
jaxatari.environment
module
jaxatari.games.jax_freeway
module
jaxatari.games.jax_kangaroo
module
jaxatari.games.jax_pong
module
jaxatari.games.jax_seaquest
module
jaxatari.spaces
module
jaxatari.wrappers
module
JAXAtariAction (class in jaxatari.environment)
JaxatariWrapper (class in jaxatari.wrappers)
JaxEnvironment (class in jaxatari.environment)
JaxFreeway (class in jaxatari.games.jax_freeway)
JaxKangaroo (class in jaxatari.games.jax_kangaroo)
JaxPong (class in jaxatari.games.jax_pong)
JaxSeaquest (class in jaxatari.games.jax_seaquest)
jump_base_y (jaxatari.games.jax_kangaroo.PlayerState attribute)
jump_counter (jaxatari.games.jax_kangaroo.PlayerState attribute)
jump_orientation (jaxatari.games.jax_kangaroo.PlayerState attribute)
just_surfaced (jaxatari.games.jax_seaquest.SeaquestState attribute)
K
KangarooConstants (class in jaxatari.games.jax_kangaroo)
KangarooInfo (class in jaxatari.games.jax_kangaroo)
KangarooObservation (class in jaxatari.games.jax_kangaroo)
KangarooRenderer (class in jaxatari.games.jax_kangaroo)
KangarooState (class in jaxatari.games.jax_kangaroo)
key (jaxatari.games.jax_pong.PongState attribute)
(jaxatari.wrappers.AtariState attribute)
L
LADDER_COLOR (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
LADDER_HEIGHT (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
ladder_positions (jaxatari.games.jax_kangaroo.KangarooObservation attribute)
(jaxatari.games.jax_kangaroo.LevelState attribute)
ladder_sizes (jaxatari.games.jax_kangaroo.LevelState attribute)
LADDER_WIDTH (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
landing_base_y (jaxatari.games.jax_kangaroo.PlayerState attribute)
lane_borders (jaxatari.games.jax_freeway.FreewayConstants attribute)
lane_dependent_pattern (jaxatari.games.jax_seaquest.SpawnState attribute)
lane_directions (jaxatari.games.jax_seaquest.SpawnState attribute)
lane_phase_offset (jaxatari.games.jax_freeway.FreewayConstants attribute)
lane_spacing (jaxatari.games.jax_freeway.FreewayConstants attribute)
lane_time (jaxatari.games.jax_freeway.FreewayState attribute)
last_stood_on_platform_y (jaxatari.games.jax_kangaroo.PlayerState attribute)
LEFT (jaxatari.environment.JAXAtariAction attribute)
LEFT_CLIP (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
LEFTFIRE (jaxatari.environment.JAXAtariAction attribute)
level (jaxatari.games.jax_kangaroo.KangarooInfo attribute)
(jaxatari.games.jax_kangaroo.KangarooState attribute)
LEVEL_1 (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
LEVEL_2 (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
LEVEL_3 (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
level_finished (jaxatari.games.jax_kangaroo.KangarooState attribute)
LevelState (class in jaxatari.games.jax_kangaroo)
levelup (jaxatari.games.jax_kangaroo.KangarooState attribute)
levelup_timer (jaxatari.games.jax_kangaroo.KangarooState attribute)
list_available_games() (in module jaxatari.core)
lives (jaxatari.games.jax_kangaroo.KangarooState attribute)
(jaxatari.games.jax_seaquest.SeaquestObservation attribute)
(jaxatari.games.jax_seaquest.SeaquestState attribute)
lives_lost (jaxatari.games.jax_freeway.FreewayState attribute)
LogState (class in jaxatari.wrappers)
LogWrapper (class in jaxatari.wrappers)
M
make() (in module jaxatari.core)
make_renderer() (in module jaxatari.core)
MAX_COLLECTED_DIVERS (jaxatari.games.jax_seaquest.SeaquestConstants attribute)
MAX_DIVERS (jaxatari.games.jax_seaquest.SeaquestConstants attribute)
MAX_ENEMY_MISSILES (jaxatari.games.jax_seaquest.SeaquestConstants attribute)
MAX_PLATFORMS (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
MAX_PLAYER_TORPS (jaxatari.games.jax_seaquest.SeaquestConstants attribute)
MAX_SHARKS (jaxatari.games.jax_seaquest.SeaquestConstants attribute)
MAX_SPEED (jaxatari.games.jax_pong.PongConstants attribute)
MAX_SUBS (jaxatari.games.jax_seaquest.SeaquestConstants attribute)
MAX_SURFACE_SUBS (jaxatari.games.jax_seaquest.SeaquestConstants attribute)
MIN_BALL_SPEED (jaxatari.games.jax_pong.PongConstants attribute)
missile_pos (jaxatari.games.jax_seaquest.CarryState attribute)
MISSILE_SIZE (jaxatari.games.jax_seaquest.SeaquestConstants attribute)
MISSILE_SPAWN_POSITIONS (jaxatari.games.jax_seaquest.SeaquestConstants attribute)
module
jaxatari.core
jaxatari.environment
jaxatari.games.jax_freeway
jaxatari.games.jax_kangaroo
jaxatari.games.jax_pong
jaxatari.games.jax_seaquest
jaxatari.spaces
jaxatari.wrappers
MONKEY_COLOR (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
MONKEY_HEIGHT (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
monkey_positions (jaxatari.games.jax_kangaroo.KangarooObservation attribute)
(jaxatari.games.jax_kangaroo.LevelState attribute)
monkey_states (jaxatari.games.jax_kangaroo.LevelState attribute)
monkey_throw_timers (jaxatari.games.jax_kangaroo.LevelState attribute)
MONKEY_WIDTH (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
MOVEMENT_SPEED (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
MultiRewardLogState (class in jaxatari.wrappers)
MultiRewardLogWrapper (class in jaxatari.wrappers)
MultiRewardWrapper (class in jaxatari.wrappers)
N
needs_release (jaxatari.games.jax_kangaroo.PlayerState attribute)
NOOP (jaxatari.environment.JAXAtariAction attribute)
NormalizeObservationWrapper (class in jaxatari.wrappers)
num_lanes (jaxatari.games.jax_freeway.FreewayConstants attribute)
O
o (jaxatari.games.jax_seaquest.PlayerEntity attribute)
ObjectCentricWrapper (class in jaxatari.wrappers)
obs_stack (jaxatari.wrappers.AtariState attribute)
(jaxatari.wrappers.PixelAndObjectCentricState attribute)
obs_to_flat_array() (jaxatari.environment.JaxEnvironment method)
(jaxatari.games.jax_freeway.JaxFreeway method)
(jaxatari.games.jax_kangaroo.JaxKangaroo method)
(jaxatari.games.jax_pong.JaxPong method)
(jaxatari.games.jax_seaquest.JaxSeaquest method)
observation_space() (jaxatari.environment.JaxEnvironment method)
(jaxatari.games.jax_freeway.JaxFreeway method)
(jaxatari.games.jax_kangaroo.JaxKangaroo method)
(jaxatari.games.jax_pong.JaxPong method)
(jaxatari.games.jax_seaquest.JaxSeaquest method)
(jaxatari.wrappers.AtariWrapper method)
(jaxatari.wrappers.FlattenObservationWrapper method)
(jaxatari.wrappers.NormalizeObservationWrapper method)
(jaxatari.wrappers.ObjectCentricWrapper method)
(jaxatari.wrappers.PixelAndObjectCentricWrapper method)
(jaxatari.wrappers.PixelObsWrapper method)
orientation (jaxatari.games.jax_kangaroo.PlayerState attribute)
oxygen (jaxatari.games.jax_seaquest.SeaquestState attribute)
OXYGEN_BAR_COLOR (jaxatari.games.jax_seaquest.SeaquestConstants attribute)
oxygen_level (jaxatari.games.jax_seaquest.SeaquestObservation attribute)
OXYGEN_TEXT_COLOR (jaxatari.games.jax_seaquest.SeaquestConstants attribute)
P
P_HEIGHT (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
PixelAndObjectCentricState (class in jaxatari.wrappers)
PixelAndObjectCentricWrapper (class in jaxatari.wrappers)
PixelAndObjectObsWrapper (class in jaxatari.wrappers)
PixelObsWrapper (class in jaxatari.wrappers)
PixelState (class in jaxatari.wrappers)
PLATFORM_COLOR (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
platform_positions (jaxatari.games.jax_kangaroo.KangarooObservation attribute)
(jaxatari.games.jax_kangaroo.LevelState attribute)
platform_sizes (jaxatari.games.jax_kangaroo.LevelState attribute)
player (jaxatari.games.jax_kangaroo.KangarooState attribute)
(jaxatari.games.jax_pong.PongObservation attribute)
(jaxatari.games.jax_seaquest.SeaquestObservation attribute)
PLAYER_ACCELERATION (jaxatari.games.jax_pong.PongConstants attribute)
PLAYER_BOUNDS (jaxatari.games.jax_seaquest.SeaquestConstants attribute)
PLAYER_COLOR (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
(jaxatari.games.jax_pong.PongConstants attribute)
(jaxatari.games.jax_seaquest.SeaquestConstants attribute)
player_direction (jaxatari.games.jax_seaquest.SeaquestState attribute)
PLAYER_HEIGHT (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
player_missile (jaxatari.games.jax_seaquest.SeaquestObservation attribute)
player_missile_position (jaxatari.games.jax_seaquest.SeaquestState attribute)
player_missile_step() (jaxatari.games.jax_seaquest.JaxSeaquest method)
player_o (jaxatari.games.jax_kangaroo.KangarooObservation attribute)
player_score (jaxatari.games.jax_pong.PongState attribute)
(jaxatari.games.jax_seaquest.SeaquestObservation attribute)
PLAYER_SIZE (jaxatari.games.jax_pong.PongConstants attribute)
(jaxatari.games.jax_seaquest.SeaquestConstants attribute)
player_speed (jaxatari.games.jax_pong.PongState attribute)
PLAYER_START_X (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
(jaxatari.games.jax_seaquest.SeaquestConstants attribute)
PLAYER_START_Y (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
(jaxatari.games.jax_seaquest.SeaquestConstants attribute)
player_step() (jaxatari.games.jax_seaquest.JaxSeaquest method)
PLAYER_WIDTH (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
player_x (jaxatari.games.jax_kangaroo.KangarooObservation attribute)
PLAYER_X (jaxatari.games.jax_pong.PongConstants attribute)
player_x (jaxatari.games.jax_seaquest.SeaquestState attribute)
player_y (jaxatari.games.jax_kangaroo.KangarooObservation attribute)
(jaxatari.games.jax_pong.PongState attribute)
(jaxatari.games.jax_seaquest.SeaquestState attribute)
PlayerEntity (class in jaxatari.games.jax_seaquest)
PlayerState (class in jaxatari.games.jax_kangaroo)
PongConstants (class in jaxatari.games.jax_pong)
PongInfo (class in jaxatari.games.jax_pong)
PongObservation (class in jaxatari.games.jax_pong)
PongRenderer (class in jaxatari.games.jax_pong)
PongState (class in jaxatari.games.jax_pong)
post_score_spawn_offset_y (jaxatari.games.jax_freeway.FreewayConstants attribute)
post_score_stun_frames (jaxatari.games.jax_freeway.FreewayConstants attribute)
prev_action (jaxatari.wrappers.AtariState attribute)
prev_sub (jaxatari.games.jax_seaquest.SpawnState attribute)
punch_counter (jaxatari.games.jax_kangaroo.PlayerState attribute)
punch_left (jaxatari.games.jax_kangaroo.PlayerState attribute)
punch_right (jaxatari.games.jax_kangaroo.PlayerState attribute)
R
range() (jaxatari.spaces.Box method)
(jaxatari.spaces.Discrete method)
(jaxatari.spaces.Space method)
render() (jaxatari.environment.JaxEnvironment method)
(jaxatari.games.jax_freeway.FreewayRenderer method)
(jaxatari.games.jax_freeway.JaxFreeway method)
(jaxatari.games.jax_kangaroo.JaxKangaroo method)
(jaxatari.games.jax_kangaroo.KangarooRenderer method)
(jaxatari.games.jax_pong.JaxPong method)
(jaxatari.games.jax_pong.PongRenderer method)
(jaxatari.games.jax_seaquest.JaxSeaquest method)
(jaxatari.games.jax_seaquest.SeaquestRenderer method)
render_object_sequentially() (jaxatari.games.jax_seaquest.SeaquestRenderer method)
RENDER_SCALE_FACTOR (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
replace() (jaxatari.wrappers.AtariState method)
(jaxatari.wrappers.LogState method)
(jaxatari.wrappers.MultiRewardLogState method)
(jaxatari.wrappers.PixelAndObjectCentricState method)
(jaxatari.wrappers.PixelState method)
RESET (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
reset() (jaxatari.environment.JaxEnvironment method)
(jaxatari.games.jax_freeway.JaxFreeway method)
(jaxatari.games.jax_kangaroo.JaxKangaroo method)
(jaxatari.games.jax_pong.JaxPong method)
(jaxatari.games.jax_seaquest.JaxSeaquest method)
(jaxatari.wrappers.AtariWrapper method)
(jaxatari.wrappers.FlattenObservationWrapper method)
(jaxatari.wrappers.LogWrapper method)
(jaxatari.wrappers.MultiRewardLogWrapper method)
(jaxatari.wrappers.NormalizeObservationWrapper method)
(jaxatari.wrappers.ObjectCentricWrapper method)
(jaxatari.wrappers.PixelAndObjectCentricWrapper method)
(jaxatari.wrappers.PixelAndObjectObsWrapper method)
(jaxatari.wrappers.PixelObsWrapper method)
reset_coords (jaxatari.games.jax_kangaroo.KangarooState attribute)
reset_level() (jaxatari.games.jax_kangaroo.JaxKangaroo method)
respawn_offset (jaxatari.games.jax_freeway.FreewayConstants attribute)
returned_episode_lengths (jaxatari.wrappers.LogState attribute)
(jaxatari.wrappers.MultiRewardLogState attribute)
returned_episode_returns (jaxatari.wrappers.LogState attribute)
(jaxatari.wrappers.MultiRewardLogState attribute)
returned_episode_returns_env (jaxatari.wrappers.MultiRewardLogState attribute)
RIGHT (jaxatari.environment.JAXAtariAction attribute)
RIGHT_CLIP (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
RIGHTFIRE (jaxatari.environment.JAXAtariAction attribute)
rng_key (jaxatari.games.jax_seaquest.SeaquestState attribute)
S
sample() (jaxatari.spaces.Box method)
(jaxatari.spaces.Dict method)
(jaxatari.spaces.Discrete method)
(jaxatari.spaces.Space method)
(jaxatari.spaces.Tuple method)
score (jaxatari.games.jax_freeway.FreewayState attribute)
(jaxatari.games.jax_kangaroo.KangarooInfo attribute)
(jaxatari.games.jax_kangaroo.KangarooState attribute)
(jaxatari.games.jax_seaquest.CarryState attribute)
(jaxatari.games.jax_seaquest.SeaquestState attribute)
SCORE_BLINK_COLORS (jaxatari.games.jax_freeway.FreewayConstants attribute)
score_blink_rate (jaxatari.games.jax_freeway.FreewayConstants attribute)
SCORE_COLOR (jaxatari.games.jax_pong.PongConstants attribute)
(jaxatari.games.jax_seaquest.SeaquestConstants attribute)
score_enemy (jaxatari.games.jax_pong.PongObservation attribute)
score_player (jaxatari.games.jax_pong.PongObservation attribute)
screen_height (jaxatari.games.jax_freeway.FreewayConstants attribute)
SCREEN_HEIGHT (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
screen_width (jaxatari.games.jax_freeway.FreewayConstants attribute)
SCREEN_WIDTH (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
SeaquestConstants (class in jaxatari.games.jax_seaquest)
SeaquestInfo (class in jaxatari.games.jax_seaquest)
SeaquestObservation (class in jaxatari.games.jax_seaquest)
SeaquestRenderer (class in jaxatari.games.jax_seaquest)
SeaquestState (class in jaxatari.games.jax_seaquest)
SHARK_DIFFICULTY_COLORS (jaxatari.games.jax_seaquest.SeaquestConstants attribute)
shark_pos (jaxatari.games.jax_seaquest.CarryState attribute)
shark_positions (jaxatari.games.jax_seaquest.SeaquestState attribute)
SHARK_SIZE (jaxatari.games.jax_seaquest.SeaquestConstants attribute)
sharks (jaxatari.games.jax_seaquest.SeaquestObservation attribute)
soft_reset_spawn_state() (jaxatari.games.jax_seaquest.JaxSeaquest method)
Space (class in jaxatari.spaces)
spawn_divers() (jaxatari.games.jax_seaquest.JaxSeaquest method)
spawn_position (jaxatari.games.jax_kangaroo.LevelState attribute)
SPAWN_POSITIONS_Y (jaxatari.games.jax_seaquest.SeaquestConstants attribute)
spawn_protection (jaxatari.games.jax_kangaroo.LevelState attribute)
spawn_state (jaxatari.games.jax_seaquest.SeaquestState attribute)
spawn_step() (jaxatari.games.jax_seaquest.JaxSeaquest method)
spawn_timers (jaxatari.games.jax_seaquest.SpawnState attribute)
SpawnState (class in jaxatari.games.jax_seaquest)
stack_space() (in module jaxatari.spaces)
step (jaxatari.wrappers.AtariState attribute)
step() (jaxatari.environment.JaxEnvironment method)
(jaxatari.games.jax_freeway.JaxFreeway method)
(jaxatari.games.jax_kangaroo.JaxKangaroo method)
(jaxatari.games.jax_pong.JaxPong method)
(jaxatari.games.jax_seaquest.JaxSeaquest method)
(jaxatari.wrappers.AtariWrapper method)
(jaxatari.wrappers.FlattenObservationWrapper method)
(jaxatari.wrappers.LogWrapper method)
(jaxatari.wrappers.MultiRewardLogWrapper method)
(jaxatari.wrappers.MultiRewardWrapper method)
(jaxatari.wrappers.NormalizeObservationWrapper method)
(jaxatari.wrappers.ObjectCentricWrapper method)
(jaxatari.wrappers.PixelAndObjectCentricWrapper method)
(jaxatari.wrappers.PixelAndObjectObsWrapper method)
(jaxatari.wrappers.PixelObsWrapper method)
step_counter (jaxatari.games.jax_kangaroo.LevelState attribute)
(jaxatari.games.jax_pong.PongState attribute)
(jaxatari.games.jax_seaquest.SeaquestInfo attribute)
(jaxatari.games.jax_seaquest.SeaquestState attribute)
step_diver_movement() (jaxatari.games.jax_seaquest.JaxSeaquest method)
step_enemy_movement() (jaxatari.games.jax_seaquest.JaxSeaquest method)
stun_frames (jaxatari.games.jax_freeway.FreewayConstants attribute)
sub_pos (jaxatari.games.jax_seaquest.CarryState attribute)
sub_positions (jaxatari.games.jax_seaquest.SeaquestState attribute)
SUBMARINE_Y_OFFSET (jaxatari.games.jax_seaquest.SeaquestConstants attribute)
submarines (jaxatari.games.jax_seaquest.SeaquestObservation attribute)
successful_rescues (jaxatari.games.jax_seaquest.SeaquestInfo attribute)
(jaxatari.games.jax_seaquest.SeaquestState attribute)
surface_sub_position (jaxatari.games.jax_seaquest.SeaquestState attribute)
surface_sub_step() (jaxatari.games.jax_seaquest.JaxSeaquest method)
surface_submarine (jaxatari.games.jax_seaquest.SeaquestObservation attribute)
survived (jaxatari.games.jax_seaquest.SpawnState attribute)
T
throw_back_frames (jaxatari.games.jax_freeway.FreewayConstants attribute)
THROWN_COCONUT_HEIGHT (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
THROWN_COCONUT_WIDTH (jaxatari.games.jax_kangaroo.KangarooConstants attribute)
time (jaxatari.games.jax_freeway.FreewayInfo attribute)
(jaxatari.games.jax_freeway.FreewayState attribute)
(jaxatari.games.jax_pong.PongInfo attribute)
timer (jaxatari.games.jax_kangaroo.LevelState attribute)
to_be_spawned (jaxatari.games.jax_seaquest.SpawnState attribute)
top_border (jaxatari.games.jax_freeway.FreewayConstants attribute)
top_path (jaxatari.games.jax_freeway.FreewayConstants attribute)
Tuple (class in jaxatari.spaces)
U
UP (jaxatari.environment.JAXAtariAction attribute)
update_enemy_spawns() (jaxatari.games.jax_seaquest.JaxSeaquest method)
update_oxygen() (jaxatari.games.jax_seaquest.JaxSeaquest method)
UPFIRE (jaxatari.environment.JAXAtariAction attribute)
UPLEFT (jaxatari.environment.JAXAtariAction attribute)
UPLEFTFIRE (jaxatari.environment.JAXAtariAction attribute)
UPRIGHT (jaxatari.environment.JAXAtariAction attribute)
UPRIGHTFIRE (jaxatari.environment.JAXAtariAction attribute)
V
vel_x (jaxatari.games.jax_kangaroo.PlayerState attribute)
W
w (jaxatari.games.jax_kangaroo.Entity attribute)
walk_animation (jaxatari.games.jax_kangaroo.PlayerState attribute)
walking_frames (jaxatari.games.jax_freeway.FreewayState attribute)
WALL_BOTTOM_HEIGHT (jaxatari.games.jax_pong.PongConstants attribute)
WALL_BOTTOM_Y (jaxatari.games.jax_pong.PongConstants attribute)
WALL_COLOR (jaxatari.games.jax_pong.PongConstants attribute)
WALL_TOP_HEIGHT (jaxatari.games.jax_pong.PongConstants attribute)
WALL_TOP_Y (jaxatari.games.jax_pong.PongConstants attribute)
width (jaxatari.games.jax_freeway.EntityPosition attribute)
(jaxatari.games.jax_pong.EntityPosition attribute)
WIDTH (jaxatari.games.jax_pong.PongConstants attribute)
width (jaxatari.games.jax_seaquest.EntityPosition attribute)
(jaxatari.games.jax_seaquest.PlayerEntity attribute)
X
x (jaxatari.games.jax_freeway.EntityPosition attribute)
(jaxatari.games.jax_kangaroo.Entity attribute)
(jaxatari.games.jax_kangaroo.PlayerState attribute)
(jaxatari.games.jax_pong.EntityPosition attribute)
(jaxatari.games.jax_seaquest.EntityPosition attribute)
(jaxatari.games.jax_seaquest.PlayerEntity attribute)
X_BORDERS (jaxatari.games.jax_seaquest.SeaquestConstants attribute)
Y
y (jaxatari.games.jax_freeway.EntityPosition attribute)
(jaxatari.games.jax_kangaroo.Entity attribute)
(jaxatari.games.jax_kangaroo.PlayerState attribute)
(jaxatari.games.jax_pong.EntityPosition attribute)
(jaxatari.games.jax_seaquest.EntityPosition attribute)
(jaxatari.games.jax_seaquest.PlayerEntity attribute)