LICENSE
README.md
pyproject.toml
memorax/__init__.py
memorax.egg-info/PKG-INFO
memorax.egg-info/SOURCES.txt
memorax.egg-info/dependency_links.txt
memorax.egg-info/requires.txt
memorax.egg-info/top_level.txt
memorax/algorithms/__init__.py
memorax/algorithms/dqn.py
memorax/algorithms/ppo.py
memorax/algorithms/pqn.py
memorax/algorithms/sac.py
memorax/buffers/__init__.py
memorax/buffers/episode_buffer.py
memorax/environments/__init__.py
memorax/environments/brax.py
memorax/environments/craftax.py
memorax/environments/environment.py
memorax/environments/gxm.py
memorax/environments/gymnax.py
memorax/environments/memory_rl.py
memorax/environments/mujoco.py
memorax/environments/navix.py
memorax/environments/popgym_arcade.py
memorax/environments/popjym.py
memorax/environments/xminigrid.py
memorax/kernels/__init__.py
memorax/kernels/linear_attention.py
memorax/loggers/__init__.py
memorax/loggers/console.py
memorax/loggers/dashboard.py
memorax/loggers/file.py
memorax/loggers/logger.py
memorax/loggers/neptune.py
memorax/loggers/tensorboard.py
memorax/loggers/wandb.py
memorax/networks/__init__.py
memorax/networks/cnn.py
memorax/networks/embedding.py
memorax/networks/feature_extractor.py
memorax/networks/heads.py
memorax/networks/identity.py
memorax/networks/mlp.py
memorax/networks/network.py
memorax/networks/vit.py
memorax/networks/blocks/__init__.py
memorax/networks/blocks/base.py
memorax/networks/blocks/ffn.py
memorax/networks/blocks/moe.py
memorax/networks/blocks/normalization.py
memorax/networks/blocks/residual.py
memorax/networks/blocks/router.py
memorax/networks/blocks/segment_recurrence.py
memorax/networks/blocks/stack.py
memorax/networks/positional_embeddings/__init__.py
memorax/networks/positional_embeddings/alibi.py
memorax/networks/positional_embeddings/base.py
memorax/networks/positional_embeddings/rope.py
memorax/networks/positional_embeddings/wpe.py
memorax/networks/sequence_models/__init__.py
memorax/networks/sequence_models/ffm.py
memorax/networks/sequence_models/linear_attention.py
memorax/networks/sequence_models/lru.py
memorax/networks/sequence_models/mamba.py
memorax/networks/sequence_models/memoroid.py
memorax/networks/sequence_models/min_gru.py
memorax/networks/sequence_models/mlstm.py
memorax/networks/sequence_models/rnn.py
memorax/networks/sequence_models/s5.py
memorax/networks/sequence_models/self_attention.py
memorax/networks/sequence_models/sequence_model.py
memorax/networks/sequence_models/shm.py
memorax/networks/sequence_models/slstm.py
memorax/networks/sequence_models/utils.py
memorax/networks/sequence_models/wrappers.py
memorax/optimizers/transforms/apply_if.py
memorax/utils/__init__.py
memorax/utils/decorators.py
memorax/utils/gae.py
memorax/utils/stats.py
memorax/utils/timestep.py
memorax/utils/transition.py
memorax/utils/typing.py
memorax/utils/update.py
memorax/utils/utils.py
memorax/utils/wrappers.py
tests/test_algorithms.py
tests/test_environments.py
tests/test_sequence_models.py