An implementation of the Reward predictive modelling approach introduced in "Reward-predictive representations generalize across tasks in reinforcement learning".