memorax.environments

Environment creation and wrappers.

memorax.environments.make(env_id, **kwargs)

Return type:

tuple

Supported Namespaces

The make function supports the following namespaces:

• gymnax:: - Gymnax environments (CartPole, Pendulum, etc.)

• brax:: - Brax physics environments (Ant, Humanoid, etc.)

• navix:: - Navigation environments

• craftax:: - Craftax multi-task environments

• popgym_arcade:: - POPGym Arcade environments

• popjym:: - POPJym memory benchmarks

• mujoco_playground:: - MuJoCo Playground environments

• xminigrid:: - XMiniGrid environments

• jaxmarl:: - JaxMARL multi-agent environments

• gxm:: - GXM environments

• grimax:: - Grimax environments

• pobax:: - POBAX environments