memorax.environments

Environment creation and wrappers.

environment.make() - Factory function to create JAX-compatible environments.

Supported Environments

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:: - MuJoCo environments

• xminigrid:: - XMiniGrid environments