The reference paper introduces a new reinforcement learning model called Attention- Gated MEmory Tagging (AuGMEnT). The results presented suggest new approaches in understanding the acquisition of tasks requiring working memory and attentional feedback, as well as biologically plausible learning mechanisms. The model also improves on previous reinforcement learning schemes by allowing tasks to be expressed more naturally as a sequence of inputs and outputs. A Python implementation of the model is available on the author’s GitHub page which helped to verify the correctness of the computations. The script written for this replication also uses Python along with NumPy.