This directory contains the core code of LeGEM. It contains the data structure for building the episodic memory and the searching function for finding the pivot timesteps. The rest Python files are code of tested environments evaluated in our paper. 

Our paper is under review. The code files are for NeurIPS review. **Please do not distribute and use our code privately**. We will release the source code once our paper is accepted.
