Episodic-Reinforcement-Learning-With-Associative-Memory-Example

This is an example of Episodic Reinforcement Learning With Associative Memory using Google's Jax library with OpenAI's Cart Pole Environment

The code is annotated to reference specific equations and algorithms in the paper.

The link for the paper: https://openreview.net/pdf?id=HkxjqxBYDB

Dependencies are in run/requirements.txt

The code is ran by running "python3 run/"