---

source

NormalizeVecReward

 NormalizeVecReward (env, gamma)

Base class for Gymnax wrappers.

---

source

NormalizeVecRewEnvState

 NormalizeVecRewEnvState (mean:jax.Array, var:jax.Array, count:float,
                          return_val:float, env_state:gymnax.environments.
                          environment.EnvState)

---

source

NormalizeVecObservation

 NormalizeVecObservation (env)

Base class for Gymnax wrappers.

---

source

NormalizeVecObsEnvState

 NormalizeVecObsEnvState (mean:jax.Array, var:jax.Array, count:float,
                          env_state:gymnax.environments.environment.EnvSta
                          te)

---

source

VecEnv

 VecEnv (env)

Base class for Gymnax wrappers.

---

source

TransformReward

 TransformReward (env, transform_reward)

Base class for Gymnax wrappers.

---

source

TransformObservation

 TransformObservation (env, transform_obs)

Base class for Gymnax wrappers.

---

source

ClipAction

 ClipAction (env, low=-1.0, high=1.0)

Base class for Gymnax wrappers.

---

source

NavixGymnaxWrapper

 NavixGymnaxWrapper (env_name)

Initialize self. See help(type(self)) for accurate signature.

---

source

BraxGymnaxWrapper

 BraxGymnaxWrapper (env_name, backend='positional')

Initialize self. See help(type(self)) for accurate signature.

---

source

LogWrapper

 LogWrapper (env:gymnax.environments.environment.Environment)

Log the episode returns and lengths.

---

source

LogEnvState

 LogEnvState (env_state:gymnax.environments.environment.EnvState,
              episode_returns:float, episode_lengths:int,
              returned_episode_returns:float,
              returned_episode_lengths:int, timestep:int)

---

source

FlattenObservationWrapper

 FlattenObservationWrapper
                            (env:gymnax.environments.environment.Environme
                            nt)

Flatten the observations of the environment.

---

source

GymnaxWrapper

 GymnaxWrapper (env)

Base class for Gymnax wrappers.