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/"