Installation

Basic Installation

Install Memorax using pip:

pip install memorax

Or using uv:

uv add memorax

Optional Dependencies

Memorax supports various environment backends. Install the ones you need:

pip install "memorax[brax]"
pip install "memorax[craftax]"
pip install "memorax[navix]"
pip install "memorax[popgym-arcade]"
pip install "memorax[popjym]"
pip install "memorax[gxm]"
pip install "memorax[jaxmarl]"
pip install "memorax[xminigrid]"
pip install "memorax[playground]"

Or install a bundled environment stack:

pip install "memorax[environments]"

GPU Support

For GPU acceleration with CUDA:

pip install "memorax[cuda]"

Development Installation

To contribute to Memorax:

git clone https://github.com/noahfarr/memorax.git
cd memorax
uv sync
uv run pre-commit install

Verifying Installation

import memorax
from memorax.algorithms import PPO
from memorax.environments import make

env, env_params = make("gymnax::CartPole-v1")
print(f"Memorax version: {memorax.__version__}")
print(f"Environment created: {env}")