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 action_space() (jaxatari.environment.JaxEnvironment method) active (jaxatari.environment.ObjectObservation attribute), [1] atari_state (jaxatari.wrappers.LogState attribute) (jaxatari.wrappers.MultiRewardLogState attribute) (jaxatari.wrappers.ObjectCentricState attribute) (jaxatari.wrappers.PixelAndObjectCentricState attribute) (jaxatari.wrappers.PixelState attribute) AtariState (class in jaxatari.wrappers) AtariWrapper (class in jaxatari.wrappers) B Box (class in jaxatari.spaces) C contains() (jaxatari.spaces.Box method) (jaxatari.spaces.Dict method) (jaxatari.spaces.Discrete method) (jaxatari.spaces.Space method) (jaxatari.spaces.Tuple method) create() (jaxatari.environment.ObjectObservation class method) D Dict (class in jaxatari.spaces) Discrete (class in jaxatari.spaces) 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 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 FIRE (jaxatari.environment.JAXAtariAction attribute) FlattenObservationWrapper (class in jaxatari.wrappers) G get_all_values() (jaxatari.environment.JAXAtariAction class method) get_object_space() (in module jaxatari.spaces) H height (jaxatari.environment.ObjectObservation attribute), [1] I image_space() (jaxatari.environment.JaxEnvironment method) (jaxatari.wrappers.AtariWrapper method) image_stack (jaxatari.wrappers.PixelAndObjectCentricState attribute) (jaxatari.wrappers.PixelState attribute) J jaxatari.core module jaxatari.environment module jaxatari.renderers module jaxatari.spaces module jaxatari.wrappers module JAXAtariAction (class in jaxatari.environment) JaxatariWrapper (class in jaxatari.wrappers) JaxEnvironment (class in jaxatari.environment) JAXGameRenderer (class in jaxatari.renderers) K key (jaxatari.wrappers.AtariState attribute) L LEFT (jaxatari.environment.JAXAtariAction attribute) LEFTFIRE (jaxatari.environment.JAXAtariAction attribute) list_available_games() (in module jaxatari.core) LogState (class in jaxatari.wrappers) LogWrapper (class in jaxatari.wrappers) M make() (in module jaxatari.core) module jaxatari.core jaxatari.environment jaxatari.renderers jaxatari.spaces jaxatari.wrappers MultiRewardLogState (class in jaxatari.wrappers) MultiRewardLogWrapper (class in jaxatari.wrappers) MultiRewardWrapper (class in jaxatari.wrappers) N NOOP (jaxatari.environment.JAXAtariAction attribute) NormalizeObservationWrapper (class in jaxatari.wrappers) O ObjectCentricState (class in jaxatari.wrappers) ObjectCentricWrapper (class in jaxatari.wrappers) ObjectObservation (class in jaxatari.environment) obs_stack (jaxatari.wrappers.ObjectCentricState attribute) (jaxatari.wrappers.PixelAndObjectCentricState attribute) observation_space() (jaxatari.environment.JaxEnvironment 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.environment.ObjectObservation attribute) P 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) preprocess_image() (in module jaxatari.wrappers) prev_action (jaxatari.wrappers.AtariState attribute) PyGameRenderer (class in jaxatari.renderers) R range() (jaxatari.spaces.Box method) (jaxatari.spaces.Discrete method) (jaxatari.spaces.Space method) render() (jaxatari.environment.JaxEnvironment method) (jaxatari.renderers.JAXGameRenderer method) replace() (jaxatari.environment.ObjectObservation method) (jaxatari.wrappers.AtariState method) (jaxatari.wrappers.LogState method) (jaxatari.wrappers.MultiRewardLogState method) (jaxatari.wrappers.ObjectCentricState method) (jaxatari.wrappers.PixelAndObjectCentricState method) (jaxatari.wrappers.PixelState method) reset() (jaxatari.environment.JaxEnvironment 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) 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) RIGHTFIRE (jaxatari.environment.JAXAtariAction attribute) S sample() (jaxatari.spaces.Box method) (jaxatari.spaces.Dict method) (jaxatari.spaces.Discrete method) (jaxatari.spaces.Space method) (jaxatari.spaces.Tuple method) Space (class in jaxatari.spaces) stack_space() (in module jaxatari.spaces) state (jaxatari.environment.ObjectObservation attribute) step (jaxatari.wrappers.AtariState attribute) step() (jaxatari.environment.JaxEnvironment 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) T Tuple (class in jaxatari.spaces) U UP (jaxatari.environment.JAXAtariAction attribute) 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 visual_id (jaxatari.environment.ObjectObservation attribute) W width (jaxatari.environment.ObjectObservation attribute), [1] X x (jaxatari.environment.ObjectObservation attribute) Y y (jaxatari.environment.ObjectObservation attribute), [1]